Staff Technical Lead Manager, Prediction & Planning, ML Eval

1 month ago
Full-time
Lead
Software Development

Waymo

Waymo is an autonomous driving technology company building the Waymo Driver and operating Waymo One, its fully autonomous ride-hailing service.

Autonomous vehicles, robotics, AI, ride-hailing / mobility tech
Founded 2009
$21600M raised

Description

  • Define the strategic vision for evaluation platforms and foundational infrastructure supporting next-generation ML model evaluation.
  • Collaborate with ML engineers, data scientists, and infrastructure teams to identify and surface critical signals for model, component, and system-level performance.
  • Scale evaluation and infrastructure platforms to improve the ML developer experience and enable faster, more trusted iteration.
  • Manage and mentor a focused team of engineers while aligning career growth with organizational needs.
  • Drive best practices and optimize evaluation workflows using deep technical knowledge of the Alphabet ML stack, including TensorFlow, JAX, Flax, and Apache Beam.
  • Stay current on emerging technologies, industry trends, and research in ML evaluation methodologies and advanced metrics design.
  • Support rigorous validation of next-generation deep neural networks used for autonomous driving planning.
  • Help accelerate ML developer velocity across the PrePlan organization.

Requirements

  • M.S. in Computer Science, Mathematics, or equivalent industry experience in Robotics or large-scale ML systems with critical evaluation needs.
  • 5+ years of experience building and maintaining large-scale distributed infrastructure, ML inference systems, or evaluation platforms.
  • 3+ years of engineering management experience.
  • Strong coding and testing proficiency in Python and C++.
  • Strong foundational knowledge of model evaluation and data science principles such as confidence intervals, outlier identification, curve fitting, and causality analysis.
  • Familiarity with large-scale ML deployment and orchestration tools such as TF Serving, TorchServe, Kubeflow, SageMaker Pipelines, or Vertex AI Pipelines.
  • Understanding of machine learning fundamentals and experience with JAX, PyTorch, or TensorFlow.
  • Experience developing and maintaining evaluation pipelines for ML models.
  • Experience deploying and supporting machine learning models for computer vision, natural language processing, robotics/motion planning, or recommendation systems.
  • Experience supporting a small team of MLEs developing high-capacity, production-grade models and components.
  • Strong understanding of metrics computation and regression detection at scale.

Benefits

  • Base salary range of $251,000 to $310,000 USD.
  • Eligibility for Waymo’s discretionary annual bonus program.
  • Eligibility for Waymo’s equity incentive plan.
  • Access to Waymo’s generous company benefits program, subject to eligibility requirements.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Lead Engineering Manager

Weekday 11-50 Construction & Engineering

Weekday’s client is hiring a Lead Engineering Manager in the USA to architect and deliver complex stream-processing systems while leading engineering teams and partnering with cross-functional stakeholders.

Agile AWS Azure CI/CD Docker GCP Java Kafka Kubernetes Scrum SQL
14 hours, 3 minutes ago

Lead Software Engineer - Factory Platform

Relativity Space 251-1K Aerospace & Defense

Relativity Space is hiring a Lead Software Engineer to guide the Factory Platform team in building software for an automated rocket factory and integrated launch platform supporting Terran R operations.

AWS dbt GCP Go Helm Kubernetes Microservices PostgreSQL Python React Terraform TypeScript
14 hours, 48 minutes ago

Software Engineer - Machine Learning (Behaviors)

Motional 1K-5K Automotive

Motional’s Behaviors team is hiring an engineer to develop machine learning models that help autonomous vehicles understand and predict traffic behavior in complex real-world driving scenarios.

C++ Computer Vision Deep Learning Machine Learning Neural Networks Python PyTorch
1 day, 13 hours ago

Engineering Manager, Attendance

SeatGeek 251-1K Media

SeatGeek is hiring a Manager to lead engineering teams and help scale the development operation behind its live events ticketing platform.

Android AWS C# Elasticsearch Git GitLab Go iOS Kafka .NET Next.js PostgreSQL Python RabbitMQ Redis Swift TypeScript
1 day, 13 hours ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers