Senior 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 distributed ML infrastructure for Waymo's autonomous fleet
Domain
Autonomous vehicles, machine learning infrastructure, large-scale simulation
Deliverable
production ML models | infrastructure
Required skills
distributed training, large-scale model training, ML system architecture, PyTorch/Jax/Tensorflow/Ray, performance profiling, gradient sharding
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; design and scale distributed systems covering the ML lifecycle; optimize ML models for fleet assessment; guide architectural decisions for complex systems
Seniority
Senior, hands-on IC