Waymo

Staff Tech Lead, Perception, Pre-training & Distillation

Waymo
Apply
2 months ago
Mountain View, CA, USA or San Francisco, CA, USAStaff+
H1B sponsor

Base Salary

$251k - $310k/yr

Responsibilities

  • Train multi-billion-parameter spatio-temporal foundation models on multi-billion-frame datasets.
  • Deliver pre-trained models for production use, ideally in real-time systems.
  • Develop self-supervised learning methods for image/video encoders, 3D point cloud encoders, spatio-temporal reconstruction or future prediction, and vision-language contrastive learning.
  • Develop multi-teacher distillation recipes.
  • Evaluate latent representation quality and collaborate with post-training teams on world knowledge representations.
  • Develop methods to curate and effectively use diverse training datasets.
  • Contribute to technical roadmaps.

Requirements

  • MS or PhD in Computer Science or a similar discipline, or an equivalent amount of deep learning experience.
  • At least 5 years of experience in Machine Learning and/or Computer Vision.
  • Fluency in Python.
  • In-depth experience with machine learning frameworks such as PyTorch or JAX.
  • Preferred: publications at top-tier conferences including CVPR, ICCV, ECCV, ICLR, ICML, ICRA, RSS, NeurIPS, AAAI, IJCV, or PAMI.
  • Preferred: experience with C++.
  • Preferred: experience with real-time system design.

Benefits

  • Eligible employees may participate in a discretionary annual bonus program and equity incentive plan.
  • Generous company benefits program, subject to eligibility requirements.
  • Full-time position; salary range and potential remote-location details vary by work location.

Categories

AI ResearchML Engineering
Waymo

About Waymo

1,001-5,000 employees

Waymo develops the Waymo Driver, a full-stack autonomous driving system, and operates the Waymo One robotaxi service that offers paid, driverless rides in select U.S. cities. The company monetizes through ride-hailing fares and partnerships integrating its technology on multiple vehicle platforms. Founded in 2009 as Google’s self-driving project, Waymo is headquartered in Mountain View and is an Alphabet subsidiary, with over 10 million rider-only trips and 100+ million autonomous miles on public roads.

Contact me