MLOps Engineer (JAX, PyTorch, Pallas/Triton)

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