Staff Machine Learning Infrastructure Engineer, Simulation

About the Company

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. The Waymo Driver powers Waymo’s fully autonomous ride-hail service and has provided over ten million rider-only trips.

Responsibilities

• Advance the state of the art of ultra realistic multi-agent simulations using foundation models
• Collaborate with Google DeepMind and Waymo Realism Modeling teams to improve simulation realism
• Provide technical leadership on large-scale ML model architectures for autonomous vehicles
• Design and scale large distributed systems covering the ML lifecycle and planet-scale dataset generation
• Translate product goals into measurable technical deliverables and mentor junior engineers

Requirements

• BS in Computer Science, Robotics, or a similar technical field
• 5+ years of professional software engineering experience
• 3+ years in machine learning infrastructure (developing, scaling, training, and deploying large-scale systems)

Preferred Qualifications

• MS in Computer Science, Robotics, or a similar technical field
• 10+ years of professional software engineering experience
• 5+ years in machine learning infrastructure
• Experience with DeepSpeed, PyTorch, or TensorFlow
• Expertise in distributed training techniques and gradient sharding
• Deep understanding of auto-regressive transformers and custom kernels
• Familiarity with Autonomous Driving, Simulations, and ML accelerators