Want to professionalize your AI skills, pivot to an AI role and increase your salary?
Master AI Engineering with the most practical and comprehensive LLM Development certifications at Towards AI Academy.

Waymo

Staff Machine Learning Engineer (Infra), Driver Understanding and Evaluation

Waymo

Published 11 May 2026
Mountain View, CA, USA
251K - 310K USD Annual
Full Time

Share this job

Role Highlights

Languages used

Key skills

AI
Data Engineer
Machine Learning
Computer Science
Integrations
Distributed Systems
Technical Leadership
Optimization
Infrastructure
API
Transformers
Robotics
Reinforcement Learning

Tools, Libraries and Frameworks

JAX
PyTorch
Tensorflow
Ray

Description

The role involves building and operating scalable machine learning and data systems alongside simulation workflows and insight tools. The individual will provide technical leadership on large-scale model architectures and oversee the production and optimization of machine learning models. Responsibilities include designing and scaling distributed systems that support dataset generation, model training, and evaluation. The position requires working at the intersection of data engineering, model development, and simulation to improve evaluation processes. Additionally, the role involves collaborating cross-functionally to translate business goals into technical deliverables.

Required Qualifications and Skills

Candidates must possess an M.S. or Ph.D. in Computer Science, Machine Learning, Artificial Intelligence, or a related field, or equivalent practical experience. A minimum of 7 years of professional software engineering experience is required, including at least 3 years specifically in machine learning infrastructure. Applicants need a history of contributions to machine learning tooling and frameworks, along with expertise in distributed training techniques and large-scale model architectures. Strong leadership skills and the ability to navigate cross-functional teams are essential for this position.

Disclaimer

Disclaimer: Job and company description information and some of the data fields may have been generated via GPT-4 summarisation and could contain inaccuracies. The full external job listing link should always be relied on for authoritative information.

About the company

Waymo

Description

Waymo is an autonomous driving technology company that began as the Google Self-Driving Car Project in 2009. With a mission to enhance mobility and safety, Waymo has developed the Waymo Driver, powering their fully autonomous ride-hailing and local delivery services. To date, Waymo has achieved over 20 million miles of autonomous driving on public roads across 25 US cities and 20 billion miles of simulation testing. The company emphasizes mission-driven innovation and a collaborative culture, aiming to improve access to mobility while prioritizing safety and community impact.

Share

Share this job

Related jobs

System Design
Reliability
Architecture
Interfaces
Mountain View, CA, USA
Full Time
Product Management
Stakeholder Management
Manager
Data
Mountain View, CA, US; San Francisco, CA, US
Full Time
Data Protection
Risk Management
Program Management
Stakeholder Management
Mountain View, CA, USA
Full Time