RemoteLeaf Logo
Log in Sign up

Staff Technical Lead Manager, Prediction & Planning, ML Eval

1 month, 3 weeks 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 72

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

x.ai

ML Infrastructure Engineer

x.ai 51-250 Internet Software & Services

SpaceXAI is hiring an ML Infrastructure Engineer to build and optimize the machine learning platform that powers recommendations on X.

United States Full-time Junior Infrastructure Engineer Machine Learning Engineer
$180k-$440k
Ansible C++ Linux Puppet Python PyTorch Rust
15 hours, 1 minute ago
Reddit

Senior Machine Learning Engineer, Safety

Reddit 1K-5K Internet Software & Services

Reddit is hiring a remote-friendly Machine Learning Engineer to build and improve safety systems that support enforcement of Reddit rules using large language models.

United States Full-time Senior Machine Learning Engineer
$217k-$303k
Deep Learning LLM Machine Learning NLP Python PyTorch TensorFlow
15 hours, 31 minutes ago
Torc

Senior, Machine Learning Engineer - 3D Perception

Torc 251-1K Road & Rail

Torc is hiring a Senior Machine Learning Engineer – 3D Perception to develop and deploy production perception models for autonomous trucks and improve Bird's Eye View understanding across its autonomy stack.

United States Full-time Senior Machine Learning Engineer
$177k-$213k
C++ Computer Vision Deep Learning Machine Learning Python PyTorch
15 hours, 46 minutes ago
Burq

Engineering Manager, Merchants

Burq 11-50 Air Freight & Logistics

Burq is hiring an Engineering Manager to lead its Merchants engineering function across two teams responsible for the merchant-facing product as the company expands into enterprise delivery.

United States Canada Full-time Lead Engineering Manager
AWS Node.js PostgreSQL TypeScript
16 hours, 1 minute 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