RemoteLeaf Logo
Log in Sign up

Staff Technical Lead Manager, Prediction & Planning, ML Eval

1 month ago
United States
Full-time
Lead
Engineering Manager
Software Development
C++ Computer Vision Kubeflow Machine Learning NLP Python PyTorch TensorFlow
Apply Now

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
View All Jobs 75

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

Weekday

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.

United States Full-time Lead Engineering Manager
Agile AWS Azure CI/CD Docker GCP Java Kafka Kubernetes Scrum SQL
14 hours, 3 minutes ago
Relativity Space

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.

United States Full-time Lead Engineering Manager Software Engineer
$157k-$235k
AWS dbt GCP Go Helm Kubernetes Microservices PostgreSQL Python React Terraform TypeScript
14 hours, 48 minutes ago
Motional

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.

Singapore Full-time Senior Machine Learning Engineer
C++ Computer Vision Deep Learning Machine Learning Neural Networks Python PyTorch
1 day, 13 hours ago
SeatGeek

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.

United States Full-time Senior Engineering Manager
$171k-$248k
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