Overview
Job Description
JAX Expert – Contract
Location: Remote (San Francisco, CA)Position: JAX ExpertType: ContractCompensation: $70–$110/hourCommitment: 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, 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 at scale.Experience writing or optimizing custom GPU kernels using Pallas or Triton.Demonstrable career progression.Ability to engage reliably for at least 40 hours/week during weekdays.Strong written communication skills and the ability to explain complex technical decisions clearly.
Application Process
Upload resumeAI interview based on your resumeSubmit formTakes 20–30 mins to complete
Resources & Support
Interview process and platform info: https://talent.docs.mercor.com/welcomeSupport: 
#hiringmercor
Company:
ChatGPT Jobs
Qualifications:
Language requirements:
Specific requirements:
Educational level:
Level of experience (years):
Senior (5+ years of experience)
