Staff Machine Learning Engineer, Simulation

The Driver Understanding and Evaluation team at Waymo develops a rich understanding of Waymo Driver's behavior. With over 1 million driverless miles per week, it is critical that Waymo can understand and assess the behavior of all its vehicles - both in the field and in simulation - with automated algorithms. The learned metrics team is a strategic bet to use machine learning to ensure we can scale to meet Waymo's goals. We collaborate across teams to bring ML to production systems and build what is Waymo's reward function. We build and operate large-scale machine learning and data systems, simulation workflows, and insight tools. We combine expert human judgements and advanced machine learning models to deliver training and evaluation data for the Waymo driver. We are looking for researchers and software engineers who are passionate about developing production grade machine learning systems for our autonomous vehicles and have an incessant drive to improve the performance of our technology stack.

This role follows a hybrid work schedule and reports to an Engineering Manager.

You will:

• Develop ML models that assess our autonomous vehicle's behavior.


• Develop ML infrastructure to support performant models.


• Collaborate across teams to bring state-of-the-art to production.

You have:

• BS in Computer Science, Robotics, Statistics, Physics, Math or another quantitative area, or equivalent work experience


• 4+ years of experience building productionized ML models


• Code and design skills: comfort building production systems (Python / C++)


• Background in applied Deep Learning


• A track record in improving model quality

We prefer:

• 6+ years of experience building productionized ML models


• Experience in reinforcement learning, transfer learning, or learning.


• Experience with large scale data and models