Lead ML Systems Engineer: Large-Scale Training & Evaluation

Neura Market; Mountain View (CA); Hybrid; USD 251,000 - 310,000; Full time

Job summary

Waymo in Mountain View, CA is seeking an experienced researcher and software engineer for the DUE Machine Learning team. The role focuses on building and operating scalable machine learning and data systems, simulation workflows, and evaluation tools to support the Waymo Driver across millions of miles.

Applicants should have 7+ years in software engineering with ML infra, a MS/PhD or equivalent, and strong experience with PyTorch, Jax, TensorFlow, or Ray.

Qualifications

• M.S. or Ph.D. in Computer Science, Machine Learning, AI, or related field, or equivalent practical experience.
• 7+ years of professional software engineering with at least 3 years in ML infrastructure (training, deploying, optimizing large-scale ML systems).
• Contributions to ML tooling/frameworks (PyTorch, Jax, TensorFlow, Ray) with strong API and internal understanding.
• Strong expertise in distributed training techniques and scaling across ML accelerators.
• Deep understanding of autoregressive transformer models.

Responsibilities

• Provide deep technical leadership on large-scale ML model architectures, especially for autonomous vehicle models.
• Build scalable systems for training and fine-tuning large-scale models to evaluate driving behaviors.
• Work at the intersection of data engineering, model development, and simulation; guide architectural decisions and directions.
• Own large, complex systems and drive architectures to meet technical and business objectives.
• Oversee production and optimization of ML models for Waymo’s fleet across millions of miles.
• Design and scale large distributed ML lifecycles supporting planet-scale dataset generation, training and evaluation.
• Collaborate cross-functionally to derive performance and system-level requirements for large ML systems.

Skills

Distributed training

Autoregressive models

Leadership

Cross-functional collaboration

ML system optimization

Education

M.S./Ph.D. in CS/ML/AI

Tools

PyTorch

Jax

TensorFlow

Ray

Job description

Waymo in Mountain View, CA is seeking an experienced researcher and software engineer for the DUE Machine Learning team. The role focuses on building and operating scalable machine learning and data systems, simulation workflows, and evaluation tools to support the Waymo Driver across millions of miles.

Applicants should have 7+ years in software engineering with ML infra, a MS/PhD or equivalent, and strong experience with PyTorch, Jax, TensorFlow, or Ray.