About the job
Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. The Driver Understanding and Evaluation (DUE) team at Waymo is developing rich metrics for understanding the behavior of the Waymo Driver in the real world, and technologies such as context and scene analysis to understand driving, understanding and augmenting real world driving data to generate rare driving events, build large scale data infrastructure, improve components such as agents and a realistic simulator. The DUE Machine Learning team will build and operate scalable machine learning and data systems, simulation workflow and insight tools, improve and speed up the evaluation and onboard developer journeys.
Responsibilities
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.
Qualifications
Minimum
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
Preferred
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