MLOps Engineer (JAX, PyTorch, Pallas/Triton)

About the Role

Join a leading AI lab's cutting-edge GenAI team to be at the core of the AI revolution, where your expertise fuels the development of the most advanced Large Language Models. Join a leading AI lab's cutting-edge GenAI team and help build foundational AI models from the ground up. We're seeking talented MLOps Engineers with deep, hands-on expertise in modern ML frameworks — specifically JAX, PyTorch, and kernel-level programming (Pallas/Triton). This role involves AI model training and evaluation work, including writing and assessing MLOps tasks

What You'll Do

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

You're a Good Fit If You

• Relevant professional experience in this field.
• Strong written communication skills.
• Available for the stated weekly commitment.

Role Highlights

Pay & Payout

Independent contractor engagement, fully remote. Mercor pays weekly via Stripe or Wise.

Opens Mercor in a new tab. You pay nothing; we may earn a referral fee.