jobs

Staff Machine Learning Engineer, Infrastructure

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experi

Company

Waymo

Location

Mountain View, CA, USA; San Francisco, CA, USA

Status

Open

Posted

2025-12-08T21:56:12+00:00

Waymo is hiring a Staff Machine Learning Engineer for its ML Infrastructure team. The role builds and optimizes multi-agent simulation environments and related logic in JAX and TensorFlow on TPUs, supports foundation models with model and data parallelism, and profiles accelerators to remove bottlenecks. Candidates need six or more years in software engineering, at least four in ML infrastructure, plus hands-on TPU/GPU training, distributed scaling, and transformer knowledge. Preferred experience includes reinforcement learning, sim-to-real transfer, robotics, or autonomous-driving simulation. Both listed sites are on-site.