Perception Machine Learning Engineer - Continuous Learning

Waymo; Mountain View (CA); Hybrid; USD 175,000 - 215,000; Full time

Job summary

Waymo is seeking a Perception Machine Learning Engineer to build the intelligent systems that see the world and shape the future of autonomous travel.

Within the Perception team, you will tackle open-ended challenges, design scalable data pipelines, apply active learning, and develop evaluation methods to ensure robust model performance. This hybrid role reports to a Technical Lead Manager.

Qualifications

• Bachelor's degree in Machine Learning, Robotics, or Computer Science.
• 3+ years in ML and/or Computer Vision.
• Proven hands-on experience applying active learning in production.

Responsibilities

• Architect Infrastructure: design and scale data pipelines for sensor data.
• Drive Model Improvement: deploy active learning for continuous model updates.
• Ensure Model Quality: develop evaluation methods and detect regressions.
• Optimize Data Efficiency: automated data curation, pruning, and downsampling.

Skills

Active learning

Data pipelines

Python

PyTorch

JAX

C++

Software engineering

Education

Bachelor's degree in Machine Learning, Robotics, or Computer Science

Tools

PyTorch

JAX

C++

Job description

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 mobility while saving thousands of lives now lost to traffic crashes. The Waymo Driver powers Waymo’s fully autonomous ride-hail service and can also be applied to a range of vehicle platforms and product use cases. The Waymo Driver has provided over ten million rider-only trips, enabled by its experience autonomously driving over 100 million miles on public roads and tens of billions in simulation across 15+ U.S. states.

As a Perception Machine Learning Engineer, you will build the intelligent systems that "see" the world, directly shaping the future of autonomous travel.

Within the Perception team, we are tackling some of the most complex, open-ended challenges in autonomous driving. Our models must constantly adapt and improve as our fleet encounters the vast, unpredictable realities of public roads. We are looking for a Machine Learning Engineer to help design and build the automated, closed-loop systems that drive this continuous improvement.

In this role, you will be the bridge between model architecture and large-scale data infrastructure. You will leverage active learning and sophisticated data curation strategies to ensure our perception models are always learning from the most informative examples. Crucially, this means managing the entire lifecycle of our data: intelligently selecting novel scenarios from the fleet while continuously pruning our existing corpus to maximize training efficiency.

In this hybrid role you will report to a Technical Lead Manager.

You will:

• Architect Infrastructure: Design and scale the data pipelines needed to mine, ingest, and manage massive volumes of sensor data from our fleet.
• Drive Model Improvement: Deploy active learning algorithms to continuously identify and select the most impactful data for training, ensuring our large models continuously adapt to new environments with incremental updates.
• Ensure Model Quality: Develop methods and recipes for evaluating real-world performance of our models, and detecting regressions in model updates. Develop and maintain ground-truth free performance metrics.
• Optimize Data Efficiency: Conduct large-scale experiments focused on data balancing, subset selection, and label quality optimization. Lead automated curation strategies—including smart pruning and downsampling—to minimize dataset bloat and maximize compute efficiency.
• Solve Long-Tail Challenges: Develop robust mining, training and evaluation pipelines for rare, safety-critical real-world scenarios.
• Innovate with Model Signals: Utilize uncertainty estimation, confidence scores, and embedding space analysis to uncover model blind spots and guide automated data acquisition.
• Collaborate Cross-Functionally: Work closely with researchers and operations teams to iterate on the end-to-end model development lifecycle.

You Have:

• A bachelor’s degree in Machine Learning, Robotics, or Computer Science.
  3+ years of professional experience in Machine Learning and/or Computer Vision.
• Proven, hands‑on experience applying active learning in production environments.
• Strong expertise in building large‑scale ML data pipelines (mining, extraction, auto‑labeling, ingestion).
• Deep understanding of data curation—balancing, core set selection, and sampling—to optimize model performance.
• Proficiency in Python and deep learning frameworks (PyTorch or JAX).
• Strong software engineering skills for writing robust, production‑ready code.

We Prefer:

• An advanced degree (MS or PhD) in Machine Learning, Robotics, or Computer Science.
• A record of publications at top‑tier conferences (e.g., CVPR, ICCV, ECCV, ICML, ICLR, NeurIPS, IROS, RSS, AAAI, IJCV, PAMI).
• Experience with C++
• Experience building data‑centric infrastructure from the ground up to accelerate model iteration cycles.

The expected base salary range for this full‑time position across US locations is listed below. Actual starting pay will be based on job‑related factors, including exact work location, experience, relevant training and education, and skill level. Your recruiter can share more about the specific salary range for the role location or, if the role can be performed remote, the specific salary range for your preferred location, during the hiring process.

Waymo employees are also eligible to participate in Waymo’s discretionary annual bonus program, equity incentive plan, and generous Company benefits program, subject to eligibility requirements.

Salary Range

$175,000—$215,000 USD