Staff Tech Lead, Perception, Pre-training & DistillationActive$310K

The opportunity

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver™—to improve access to…

What you'll do

  • Train multi-billion parameter spatio-temporal foundation models on multi-billion frame data sets.

  • Deliver pre-trained models for production usage, ideally for usage in real-time systems.

  • Develop self-supervised learning, including image/video encoder, 3D point: cloud encoder, spatio-temporal reconstruction or future prediction, and vision-language contrastive learning.

  • Develop multi-teacher distillation recipes.

  • Develop methods to understand and evaluate the quality of latent: representations from pre-training. Collaborate with the post-training team to ensure pre-training delivers the world knowledge representation needed for the multitude of post-training tasks.

  • Develop methods to curate and effectively use diverse datasets in training.

What they're looking for

  • Be a key contributor to technical roadmaps.
  • MS or PhD in Computer Science or a similar discipline, or an equivalent amount of deep learning experience
  • + years experience in Machine Learning and/or Computer Vision
  • Fluent in Python