Staff Tech Lead, Perception, Pre-training & Distillation

Highlights

• Train multi-billion-parameter spatiotemporal foundation models on massive data.
• Build perception systems powering millions of fully autonomous rider trips.
• Deliver pre-trained models for real-time use in the Waymo Driver.

What they offer

• Discretionary annual bonus program
• Equity incentive plan
• Company benefits program

Ideal candidate

A research-driven ML leader who loves learning from massive real-world data and turning foundation models into real driving impact. You thrive owning ambiguous roadmaps and collaborating across teams.

Requirements

• MS or PhD in Computer Science or similar discipline, or equivalent deep learning experience
• 5+ years experience in Machine Learning and/or Computer Vision
• Fluent in Python
• In-depth experience with ML frameworks like PyTorch or JAX
• Publications at top-tier conferences preferred
• Experience with C++ preferred
• Experience with real-time system design preferred