About the job
Waymo is at the forefront of autonomous driving technology, dedicated to becoming the world's most trusted driver. Originally initiated as the Google Self-Driving Car Project in 2009, our team has been relentlessly working to enhance mobility access and save lives by developing the Waymo Driver—The World’s Most Experienced Driver™. With over ten million successful rider-only trips facilitated by our technology, which has driven over 100 million miles on public roads and tens of billions of miles in simulations across more than 15 U. S. states, we are shaping the future of transportation.
The Simulation ML Infrastructure team is responsible for creating scalable AI/ML infrastructure that propels our Simulator team towards innovative, state-of-the-art simulations of realistic environments for the Waymo Driver's training and testing. By leveraging large foundation models trained on extensive datasets, we enhance the authenticity and control of our simulations, including realistic representations of vehicles, pedestrians, cyclists, roads, traffic control systems, and weather conditions.
We are currently seeking a highly skilled Senior Machine Learning Infrastructure Engineer to spearhead the development of advanced AI/ML infrastructure for multi-billion parameter foundation models tailored for ML accelerator-friendly simulations. Your extensive knowledge in massive model scaling, ML accelerators, and distributed training will be vital in designing and optimizing our systems.

