🛠 Tech Stack

💼 About This Role

You'll provide deep technical leadership on large-scale ML model architectures for autonomous vehicles at Waymo. Your work will directly impact the performance of the world's most experienced driver across millions of miles. You'll own complex systems that combine expert human judgments with advanced ML models to deliver training and evaluation data.

🎯 What You'll Do

• Design and scale large distributed systems for the ML lifecycle.
• Build scalable systems for training and fine-tuning large-scale models.
• Drive architectures that meet technical and business objectives.
• Oversee production and optimization of ML models for fleet evaluation.

📋 Requirements

• M.S. or Ph.D. in Computer Science, ML, AI, or related field.
• 5+ years of professional software engineering, with 3+ years in ML infrastructure.
• Experience with PyTorch, Jax, Tensorflow, or similar frameworks.
• Deep understanding of distributed training techniques and optimization strategies.

✨ Nice to Have

• 7+ years of software engineering with 5+ years in ML infrastructure.
• Experience in autonomous vehicles, robotics, or simulation.
• Familiarity with RL techniques for fine-tuning large models.

🎁 Benefits & Perks

• 💰 Competitive base salary between $238,000-$302,000.
• 📈 Equity incentive plan and discretionary annual bonus.
• 🏥 Generous company benefits program (specifics not listed).