Machine Learning Engineer, Driver Understanding and Evaluation

We are looking for researchers and software engineers who are passionate about developing machine learning techniques for the Evaluation systems on our autonomous vehicles, and have an incessant drive to improve the performance of our technology stack.

- Drive innovation in ultra-realistic world simulation for autonomous vehicles. You will develop the latest simulation technologies using foundation models for embodied agents.

- Apply your deep Machine Learning (ML) expertise, in Generative AI (GenAI), to push the boundaries of simulation realism and directly influence the advancement of autonomous driving technology.

- Work with Waymo's foundational AI research team to transfer research into scalable and production-ready solutions.

- 3+ years of experience with ML software engineering

- 3+ years of experience with Python

- 3+ years of ML software experience with machine learning framework including Pytorch, Tensorflow, Jax/Flax ,

- Experience working with, creating and/or developing ML models for robotics and/or self driving vehicles

- Strong C++ in production setting, with experience in integrating ML models in production system

- Experience in training large scale models on GPU/TPU clusters are strongly preferred

#LI-Hybrid

The expected base salary range for this full-time position across US locations is listed below. Actual starting pay will be based on job-related factors, including exact work location, experience, relevant training and education, and skill level. Your recruiter can share more about the specific salary range for the role location or, if the role can be performed remote, the specific salary range for your preferred location, during the hiring process.

Waymo employees are also eligible to participate in Waymo's discretionary annual bonus program, equity incentive plan, and generous Company benefits program, subject to eligibility requirements.

$170,000 - $216,000 USD