Senior Machine Learning Engineer (Infra)
Core
Build scalable distributed systems for training, fine-tuning, and evaluating large-scale machine learning models that assess driving behaviors for Waymo's vehicle fleet.
Role type
Senior Machine Learning Infrastructure Engineer
Builds
Distributed ML training and evaluation systems for autonomous vehicle fleets
Domain
Autonomous driving / Machine Learning Infrastructure
Deliverable
production ML models
Required skills
distributed training, large-scale model optimization, ML framework development, system architecture design, performance profiling, cross-functional collaboration
Preferred skills
autonomous vehicles, robotics, complex simulation environments, large-scale simulation platforms integration
Technologies
PyTorch, JAX, TensorFlow, Ray
Responsibilities
Design and scale distributed systems for dataset generation, model training, and evaluation; Contribute to production and optimization of ML models assessing vehicle fleet behaviors; Provide guidance on architecture and technical direction for large, complex systems
Seniority
Senior, hands-on IC
