About the job
Waymo is at the forefront of autonomous driving technology, dedicated to becoming the world’s most trusted driver. Originating from the Google Self-Driving Car Project in 2009, Waymo has been relentlessly focused on creating the Waymo Driver—The World’s Most Experienced Driver™—to enhance mobility access and prevent traffic-related fatalities. The Waymo Driver is the driving force behind our fully autonomous ride-hail service, optimized for various vehicle platforms and applications. With over ten million rider-only trips and more than 100 million miles driven autonomously on public roads, alongside tens of billions of simulated miles across 15+ U. S. states, Waymo is redefining transportation.
The ML Platform team at Waymo plays a critical role by offering a suite of tools that facilitate and automate the entire machine learning workflow lifecycle, including feature and experiment management, model development, optimization, and monitoring. Our initiatives have made machine learning more accessible across diverse teams at Waymo, including Perception, Planner, Research, and Simulation.
We are seeking talented engineers with expertise in machine learning software or systems to enhance compute performance both in the cloud and on vehicles. You'll engage with the entire ML stack from a systems perspective, tackling challenges such as efficient deep learning models, model compression, and improving ML software (e.g., JAX, XLA, Triton, and CUDA). This hybrid position reports directly to the Senior Manager of Runtime and Optimization.

