Staff Machine Learning Engineer (Simulation)

TL;DR

Staff Machine Learning Engineer (Simulation): Developing and scaling production-grade machine learning models to assess autonomous vehicle behavior with an accent on large-scale data systems and simulation workflows. Focus on building reward functions, integrating state-of-the-art ML into production, and improving the performance of the autonomous driving technology stack.

Location: Onsite in Mountain View, CA; San Francisco, CA; or New York City, NY

Salary: $251,000–$310,000 USD

Company

Waymo is an autonomous driving technology company dedicated to building the world's most trusted driver to improve mobility and safety.

What you will do

• Develop and deploy ML models to evaluate autonomous vehicle behavior.
• Build and maintain robust ML infrastructure to support high-performance models.
• Collaborate across cross-functional teams to bring state-of-the-art research into production systems.
• Design and operate large-scale data systems and simulation workflows.
• Combine expert human judgment with advanced ML models to deliver high-quality training data.

Requirements

• BS in Computer Science, Robotics, Statistics, Physics, Math, or equivalent experience.
• 5+ years of experience building productionized ML models.
• Proficiency in Python and C++ for production system design.
• Strong background in applied Deep Learning.
• Proven track record in improving model quality.

Nice to have

• 8+ years of experience building productionized ML models.
• Experience in reinforcement learning or transfer learning.
• Experience working with large-scale data and complex models.

Culture & Benefits

• Participation in discretionary annual bonus program and equity incentive plan.
• Comprehensive company benefits program.
• Opportunity to work on cutting-edge autonomous driving technology.
• Collaborative environment focused on safety and innovation.