Machine Learning Engineer (Infra), Driver Understanding and Evaluation
Core
Build and operate scalable ML/data systems, simulation workflows, and evaluation tools to assess autonomous driving behaviors and onboard developers.
Role type
Senior IC machine learning infrastructure engineer (autonomous driving evaluation)
Builds
Scalable training/evaluation systems, simulation workflows, and ML lifecycle tools for Waymo's autonomous fleet
Domain
Autonomous driving / Robotics / Simulation
Deliverable
production ML models | infrastructure
Required skills
distributed training, large-scale model scaling, ML system architecture, PyTorch/Jax/Tensorflow/Ray, gradient sharding, ML accelerator profiling
Preferred skills
autonomous vehicles domain experience, robotics, complex simulation environments, large-scale simulation platform integration
Technologies
PyTorch, Jax, Tensorflow, Ray
Responsibilities
Build scalable systems for training/fine-tuning large-scale models; Provide architectural guidance for complex systems; Contribute to production/optimization of ML models for fleet assessment; Design and scale distributed systems covering the ML lifecycle; Collaborate cross-functionally to derive system requirements
Seniority
Senior, hands-on IC