Machine Learning Engineer, Driver Understanding and Evaluation

Job Summary

The Waymo Driver Understanding and Evaluation (DUE) team is seeking researchers and software engineers with a passion for developing machine learning techniques for autonomous vehicles. This role involves building and operating scalable machine learning and data systems, simulation workflow and insight tools, improving and speeding up the evaluation and onboard developer journeys.

Responsibilities

• Drive innovation in ultra-realistic world simulation for autonomous vehicles
• Develop the latest simulation technologies using foundation models for embodied agents
• Apply 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

Requirements

• 3+ years of experience with ML software engineering
• 3+ years of experience with Python
• 3+ years of ML software experience with machine learning frameworks including Pytorch, Tensorflow, Jax/Flax
• Strong C++ in production setting, with experience in integrating ML models in production system
• Experience in training large scale models on GPU/TPU clusters

Qualifications

• Passionate about developing machine learning techniques for the Evaluation systems on our autonomous vehicles
• Incessant drive to improve the performance of our technology stack

Skills

• Deep Machine Learning (ML) expertise, in Generative AI (GenAI)

Pay

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.

Salary Range $170,000—$216,000 USD