MLOps Engineer (JAX, PyTorch, Pallas/Triton)

Mercor • Mumbai Metropolitan Region
Remote
Apply
AI Summary

MLOps Engineer responsible for guiding research and engineering teams to improve AI model performance in MLOps, training infrastructure, and ML framework-level topics. Requires 2+ years of experience in ML infrastructure, MLOps, or ML systems engineering. Must have hands-on production experience with JAX and/or PyTorch at scale.

Key Highlights
Guide research and engineering teams
Design challenging tasks and write solutions
Evaluate MLOps tasks and solutions
Key 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.
Technical Skills Required
JAX PyTorch Pallas Triton
Benefits & Perks
$35-$45/hour
Remote work
40 hours/week

Job Description


About The Job

Mercor connects elite creative and technical talent with leading AI research labs. Headquartered in San Francisco, our investors include Benchmark, General Catalyst, Peter Thiel, Adam D'Angelo, Larry Summers, and Jack Dorsey.

Position: MLOps Engineer (JAX, PyTorch, Pallas/Triton) — IN

Type: Contract

Compensation: $35–$45/hour

Location: Remote

Commitment: 40 hours/week

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.


Qualifications

Must-Have

  • 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.


Compensation & Legal

  • W-2 employment with Cincinnatus LLC.
  • Paid weekly.


Application Process (Takes 20–30 mins to complete)

  • Upload resume
  • AI interview based on your resume
  • Submit form


Resources & Support

  • For details about the interview process and platform information, please check: https://talent.docs.mercor.com/welcome
  • For any help or support, reach out to: support@mercor.com


PS: Our team reviews applications daily. Please complete your AI interview and application steps to be considered for this opportunity.

Similar Jobs

Explore other opportunities that match your interests

Senior Machine Learning Engineer - Benchmark-Driven Evaluation

Machine Learning
•
2h ago

Premium Job

Sign up is free! Login or Sign up to view full details.

•••••• •••••• ••••••
Job Type ••••••
Experience Level ••••••

fetchjobs.co

India

Senior AI/ML Engineer

Machine Learning
•
2h ago
Visa Sponsorship Relocation Remote
Job Type Full-time
Experience Level Mid-Senior level

wizerr ai

United State

Senior Analyst, Predictive Analytics and Machine Learning

Machine Learning
•
7h ago

Premium Job

Sign up is free! Login or Sign up to view full details.

•••••• •••••• ••••••
Job Type ••••••
Experience Level ••••••

Credit Acceptance

United State

Subscribe our newsletter

New Things Will Always Update Regularly