Senior Machine Learning Engineer (Infra)

Home > Python programming jobs

Senior Machine Learning Engineer (Infra) in USA new

Waymo
Full time
Email

USA
10/04/2026
Report

Mountain View, CA

Responsibilities

Build scalable systems for training and fine-tuning large-scale models that evaluate driving behaviors.
Develop and operate machine learning and data systems, simulation workflows, and evaluation tools.
Design and scale distributed systems for dataset generation, model training, and evaluation across the ML lifecycle.
Contribute to production, deployment, and optimization of machine learning models for Waymo’s vehicle fleet.
Provide architectural guidance, own complex systems, and drive technical decisions.
Collaborate cross-functionally to define system requirements and translate business goals into measurable technical deliverables.

Requirements

M.S. or Ph.D. in Computer Science, Machine Learning, Artificial Intelligence, or a related technical field, or equivalent practical experience.
At least 5 years of experience developing, designing, scaling, training, deploying, and optimizing large-scale machine learning systems and infrastructure.
Contributions to machine learning tooling and frameworks such as PyTorch, Jax, Tensorflow, Ray, or similar, with understanding of user-facing APIs and internal workings.
Strong expertise in distributed training, including gradient sharding and optimization strategies for scaling large models.
Experience with ML accelerator profiling tools to identify performance bottlenecks.
Preferred: 7 or more years of machine learning infrastructure experience.
Preferred: experience in autonomous vehicles, robotics, or complex simulation environments.
Preferred: familiarity with large-scale simulation platforms and their integration with ML training workflows.

Benefits

Full-time position available across U.S. locations.
Eligibility for a discretionary annual bonus program, equity incentive plan, and company benefits program, subject to eligibility requirements.

Salary: $213k - $263k/yr

Waymo

On the journey to be the world's most trusted driver. With the Waymo Driver, we can improve mobility while saving thousands of lives. Download the Waymo One app and ride today. Waymo reaches out to candidates from official channels only (e.g. directly from @waymo.com email addresses, or through o...

Similar positions

Graduate Software Engineer

Synack
Full time

UK
10/04/2026

Remote

Software Engineer - Cloud Images

Canonical
Full time

Remote
10/04/2026

EMEA, Americas

GTM Engineer

Higgsfield
Full time

USA
10/04/2026
Salary: $130k - $175k/yr

San Francisco, CA

Research Engineer

Clera
Full time

Others
10/04/2026

Zürich, Switzerland

Lead Automation Quality Engineer

London Stock Exchange Group
Full time

Others
10/04/2026
Salary: Competitive

Bengaluru, India