RemoteLeaf Logo
Log in Sign up

Staff Technical Lead Manager, Prediction & Planning, ML Eval

19 minutes 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 77

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

instacart.careers

Senior Machine Learning Engineer II, Ads Response Prediction

instacart.careers 1K-5K Internet Software & Services

Instacart is hiring a Senior Machine Learning Engineer II to lead research and development of ads response prediction models for its ads ecosystem, focusing on model quality, debiasing, and next-generation ranking and retrieval methods.

United States Full-time Senior Machine Learning Engineer
$201k-$254k
Apache Spark dbt LLM Pandas Python PyTorch SQL TensorFlow
19 minutes ago
Anduril Industries

Test and Evaluation Engineer - Mission Autonomy

Anduril Industries 1K-5K Aerospace & Defense

Anduril Industries is hiring a Mission Autonomy Test & Evaluation Engineer to lead testing programs for autonomous defense systems and ensure mission autonomy capabilities meet performance and reliability standards.

United States Full-time Mid Level Machine Learning Engineer Test Engineer
$113k-$149k
C++ Python
34 minutes ago
Anduril Industries

Manager, Product Security Foundations

Anduril Industries 1K-5K Aerospace & Defense

Anduril Industries is hiring a Manager of Product Security Foundations to lead a security engineering team delivering reusable foundational security components for its defense products.

United States Full-time Lead Application Security Engineer Engineering Manager
$191k-$253k
C C++ Cybersecurity Embedded Systems Go IoT Linux Python Rust
34 minutes ago
instacart.careers

Engineering Manager, Enterprise Platform

instacart.careers 1K-5K Internet Software & Services

Instacart is hiring an Engineering Manager in Toronto to lead the Enterprise Platform team building the APIs, tools, and data models behind its Fulfillment-as-a-Service offering for retailers.

United States Full-time Lead Engineering Manager
$192k-$242k
AWS CI/CD E-commerce GCP Go gRPC Java Python REST API
36 minutes 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