Tech Lead Manager ML Optimization
Core
Leading the development and efficient deployment of large-scale machine learning models for autonomous driving, focusing on model system codesign and performance optimization across datacenter and on-device platforms.
Role type
Senior Tech Lead Manager, ML Infrastructure
Builds
Efficient ML infrastructure and deployment pipelines for Waymo's autonomous driving stack (Perception and Trajectory planning)
Domain
Autonomous driving, Machine Learning Infrastructure, High-performance computing
Deliverable
production ML models
Required skills
Large-scale ML system development, Model optimization, Performance profiling, Cross-functional technical leadership, Team management (10+ engineers), Distributed training/inference, Custom kernel development
Preferred skills
Master's or PhD in CS/Engineering
Technologies
DeepSpeed, PyTorch, TensorFlow, JAX, TPUs
Responsibilities
Study and translate SOTA model architectures (World Models, Diffusion) into technical deliverables; Develop tooling for model performance inspection and roofline analysis; Innovate high-performance optimizations for large-scale training/inference; Guide cross-team integration of data generation, model development, and deployment; Mentor junior engineers and manage IC performance.
Seniority
Senior, hands-on IC with management responsibilities