Machine Learning Engineer, Runtime & Optimization
Core
Improving compute performance for Waymo's autonomous driving ML models on both cloud and self-driving cars.
Role type
Senior ML systems engineer (runtime & optimization)
Builds
Efficient deep learning models and ML software for Waymo Driver
Domain
Autonomous driving / ML systems
Deliverable
production ML models
Required skills
ML system architecture, performance analysis, Python, C++, model optimization, efficient deep learning, runtime libraries, ML compilers
Preferred skills
GPU/TPU hardware architecture, foundation model optimization, GPU/TPU system software
Technologies
JAX, XLA, Triton, CUDA, PyTorch, TensorFlow
Responsibilities
Lead collaboration with ML scientists to identify system and model optimization opportunities; Develop junior engineers and lead projects from proposal to execution; Analyze and improve ML system workloads on cloud and vehicles; Apply model optimization and efficient deep learning techniques to ML systems
Seniority
Senior, technical lead