jobs

Senior Machine Learning Engineer, Runtime and Serving

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

Status

Open

Posted

2026-04-29T05:12:37+00:00

Waymo's ML Optimization team is hiring a senior engineer to build the company's onboard ML inference engine and large-scale serving systems. The work spans JAX-native runtimes, ML compiler and runtime changes (OpenXLA/PjRT, TensorRT), profiling and benchmarking tooling, and hardware-aware optimization for GPUs, TPUs or custom silicon, balancing real-time onboard limits against offboard throughput. The posting asks for a B.S. or M.S., 5+ years in ML systems and C++, 3+ years in Python with PyTorch or JAX, and experience with accelerators and low-latency distributed backends. The listed base range is $213,000–$263,000.