Waymo

Staff Machine Learning Engineer, Tech Lead, Labeling Automation

Waymo
Apply
2 months ago
Mountain View, CA, USAStaff+
H1B sponsor

Base Salary

$251k - $310k/yr

Responsibilities

  • Architect and deploy scalable auto-labeling pipelines to increase data throughput and reduce reliance on manual annotation.
  • Build automated anomaly detection and auto-grading systems to evaluate annotation quality, detect regressions, and enforce quality standards across millions of labels.
  • Train, optimize, and deploy 2D and 3D computer vision models for object detection, segmentation, and spatial tracking.
  • Fine-tune and deploy vision-language and large language models for open-set labeling and contextual reasoning tasks using prompt optimization and post-training techniques.
  • Set technical strategy, guide architectural decisions, and mentor senior and mid-level engineers.
  • Collaborate with Perception, Planner, and Simulation teams to align labeling capabilities with Waymo Driver data needs.

Requirements

  • 8+ years of professional experience in software engineering and applied machine learning.
  • Programming experience in C++ or Python.
  • Experience building, evaluating, and deploying deep learning models for object detection, segmentation, and spatial tracking.
  • Experience with large-model training, distributed computing, and scaling deep learning architectures using PyTorch or TensorFlow.
  • Experience taking machine learning solutions from research and experimentation through robust, scaled production deployment.
  • Preferred: experience building internal tooling for machine learning developers.

Benefits

  • Hybrid work arrangement.
  • Eligibility for a discretionary annual bonus program, equity incentive plan, and company benefits program, subject to eligibility requirements.
  • Full-time position with a stated U.S. base salary range of $251,000–$310,000.

Tech Stack

Categories

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