MLOps Engineer for AI Model Development

Join a cutting-edge GenAI team at a leading AI lab and play a pivotal role in the development of advanced Large Language Models. This position is ideal for MLOps Engineers with extensive hands-on experience in modern ML frameworks, specifically JAX, PyTorch, and kernel-level programming (Pallas/Triton).

• Guide research and engineering teams to close knowledge gaps and enhance 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, providing 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 consistenc...