Skip to main content

About the role

Waymo is hiring a Staff Machine Learning Engineer, AI Infrastructure, to lead the design and implementation of scalable ML infrastructure. You will work across data generation, model training, and production serving, ensuring reliability and performance for real-world autonomous driving data and simulations. The role requires 7+ years of hands-on ML systems experience, strong Python/C++ coding, and expertise with JAX/TensorFlow/PyTorch.

Matching similar jobs

JOB OVERVIEW

Experience level

Lead

Location

Millbrae, CA

Occupation

Computer Systems Engineers/Architects

Industry

Computer Systems Design Services

Posted

yesterday

Tired of running searches?

Rank the roles you'd take once, and matches like these arrive on their own.

CREATE PROFILE (opens in a new tab)