Senior Machine Learning Engineer (Infra), Driver Understanding and Evaluation

Improve autonomous vehicle technology through advanced ML techniques.

• •Waymo is seeking a Senior Machine Learning Engineer to build and operate scalable machine learning and data systems for driver understanding and evaluation.
• •This role focuses on improving the performance of autonomous vehicle technology through advanced ML techniques.
• •Key Responsibilities Build scalable systems for training and fine-tuning large-scale models to evaluate driving behaviors.
• •Work at the intersection of data engineering, model development, and simulation.
• •Design and scale large distributed systems covering the ML lifecycle.
• •Collaborate cross-functionally to derive performance and system-level requirements for large ML systems.
• •Requirements M.S. or Ph.D. degree in Computer Science, Machine Learning, AI, or related field, or equivalent practical experience. 5+ years in machine learning infrastructure, including developing, designing, scaling, training, deploying, and optimizing large-scale ML systems.
• •Contributions to machine learning tooling and frameworks like PyTorch, Jax, Tensorflow, or Ray.
• •Strong expertise in distributed training techniques and ML accelerator profiling tools.