Senior ML Infra Engineer - Driver Evaluation & Simulation

Waymo; Mountain View (CA); On-site; USD 251,000 - 310,000; Full time

Benefits offered by this job

Annual bonus program

Equity incentive plan

Generous Company benefits program

Job summary

Waymo in Mountain View, CA is seeking a Machine Learning specialist to lead large-scale model development for autonomous vehicles. This role requires an M.S. or Ph.D. in Computer Science, significant software engineering experience, and deep expertise in machine learning infrastructure. The expected salary ranges from $251,000 to $310,000 USD, with eligibility for a bonus program and comprehensive benefits. Ideal candidates will have a track record in ML tooling and leadership across teams.

Qualifications

• 7+ years of software engineering experience, with 3+ years in machine learning infrastructure.
• History of contributions to machine learning tooling and frameworks.
• Expertise in distributed training techniques for large models.
• Strong leadership skills with experience in cross-functional teams.

Responsibilities

• Provide deep technical leadership on large-scale ML model architectures.
• Build scalable systems for training and fine-tuning models for driving behaviors.
• Oversee the production and optimization of machine learning models for vehicles.
• Collaborate cross-functionally to derive performance and system-level requirements.

Skills

Machine Learning Techniques

Software Engineering

Data Engineering

Distributed Training Techniques

Technical Leadership

Robotics Knowledge

Education

M.S. or Ph.D. in Computer Science or related field

Tools

PyTorch

Tensorflow

Jax

Ray

Job description

Waymo in Mountain View, CA is seeking a Machine Learning specialist to lead large-scale model development for autonomous vehicles. This role requires an M.S. or Ph.D. in Computer Science, significant software engineering experience, and deep expertise in machine learning infrastructure. The expected salary ranges from $251,000 to $310,000 USD, with eligibility for a bonus program and comprehensive benefits. Ideal candidates will have a track record in ML tooling and leadership across teams.