About the job
Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. The Simulator Team at Waymo builds state-of-the-art simulations of realistic environments for testing and training the Waymo Driver. We develop industry-leading simulation solutions using advanced ML algorithms that measure and enhance the performance of the Waymo Driver. By applying machine learning, we model the real world, including realistic agents (vehicles, pedestrians, cyclists, motorcyclists), roads, traffic control systems, and weather conditions.
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
5+ years of experience with ML software engineering
5+ years of experience with Python
5+ 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