Machine Learning Ops Engineer (JAX, PyTorch, Pallas/Triton)

Great Value Hiring

Data Science Analytics & Machine Learning

149+ Applicants

Posted: 1 month ago

2-4 years

India

work from office

Posted: 1 month ago

Applicants: 149+

Job Description

Similar Jobs

Please verify your account first! Send OTP

Please click on the Apply to verify the status of jobs posted more than 15 days ago, as they may have expired. Similar Jobs

Job Description

MLOps Engineer (JAX, PyTorch, Pallas/Triton) [$35-$45/hr] [W2, Contingent Role]
Talented MLOps Engineers with deep, hands-on expertise in modern ML frameworks specifically JAX, PyTorch, and kernel-level programming (Pallas/Triton)
This is a W-2 employment position requiring a commitment of 40 hours per week (during weekdays).
Role Responsibilities

Guide research and engineering teams to close knowledge gaps and improve AI model performance in MLOps, training infrastructure, and ML framework-level topics
Design challenging, domain-relevant tasks, and write accurate and well-structured solutions to MLOps and ML systems problems
Evaluate MLOps tasks and solutions and provide clear, written technical feedback
Develop guidelines and detailed rubrics/evaluation frameworks to assess training pipeline design, distributed systems reasoning, and kernel-level optimization across tasks
Collaborate with other subject matter experts to ensure consistency and accuracy in training data

Good Candidature

Looking to get Placed? Try our Placement Guarantee Plan

2+ years of dedicated professional experience in ML infrastructure, MLOps, or ML systems engineering at a recognized, top-tier organization
Hands-on production experience with JAX and/or PyTorch at scale
Experience writing or optimizing custom GPU kernels using Pallas (JAX) or Triton
Demonstrable career progression
Ability to engage reliably for at least 30 hours/week during weekdays
Strong written communication skills and the ability to explain complex technical decisions clearly

Skills

AiMlMachine Learning

If a job posting appears fraudulent, asks for payment, contains misleading information, or violates our guidelines, please report it immediately. Our team will review it promptly, Jobaaj does not charge any fee from the applicants.

Important dates & deadlines?

Application Deadline

15 Jul 26, 06:21 PM IST

Similar Jobs

View All

Jobaaj

Don't Miss out any Updates

Subscribe now for the latest job alerts
and never miss an update

Job Alert

Google hiring for Specific Roles Apply Now!

1 min ago

New Opportunity

Amazon is hiring freshers Apply Now!

5 min ago

Featured Jobs

Microsoft opening 50+ positions Apply Now!

10 min ago