Senior ML Infra Engineer - Ultra-Realistic Simulation

Waymo; San Francisco (CA); On-site; USD 213,000 - 263,000; Full time

Benefits offered by this job

Discretionary annual bonus program

Equity incentive plan

Generous company benefits program

Job summary

Waymo is seeking a Senior Machine Learning Infrastructure Engineer to lead the development of AI/ML infrastructure for advanced simulations. Candidates should have significant experience in software engineering and ML infrastructure, focusing on large models.

This role supports a high-performing engineering team and involves collaboration with Core Google DeepMind teams for simulation realism. Competitive salary and benefits available.

Qualifications

• 5+ years of professional software engineering experience with ML infrastructure.
• Strong expertise in distributed training and large-scale ML systems.

Responsibilities

• Lead development of advanced AI/ML infrastructure for simulations.
• Collaborate with teams to improve simulation realism.
• Mentor junior engineers and foster collaborative culture.

Skills

Machine Learning Infrastructure

Software Engineering

Distributed Training Techniques

Model Optimization

Education

BS in Computer Science or similar

Tools

DeepSpeed

PyTorch

TensorFlow

Job description

Waymo is seeking a Senior Machine Learning Infrastructure Engineer to lead the development of AI/ML infrastructure for advanced simulations. Candidates should have significant experience in software engineering and ML infrastructure, focusing on large models.

This role supports a high-performing engineering team and involves collaboration with Core Google DeepMind teams for simulation realism. Competitive salary and benefits available.