Technical Lead Manager, Prediction, ML Evaluation

About the Company

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. The Waymo Driver powers Waymo’s fully autonomous ride-hail service and has provided over ten million rider-only trips.

Responsibilities

• Influence the strategic direction of foundational infrastructure and evaluation platforms
• Collaborate cross-functionally with ML engineers, data scientists, and infrastructure teams
• Scale evaluation and infrastructure platforms to enhance the ML developer experience
• Manage and mentor a focused team of engineers
• Drive best practices using the Alphabet ML stack including TensorFlow, JAX, Flax, and Apache Beam
• Stay at the forefront of emerging technologies and research in ML evaluation methodologies

Requirements

• M.S. in Computer Science, Mathematics, or equivalent industry experience in Robotics or large-scale ML systems
• 5+ years of experience building 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 core data science principles
• Familiarity with large-scale ML deployment and orchestration tools like TF Serving, Kubeflow, or Vertex AI
• Understanding of machine learning fundamentals and frameworks such as JAX, PyTorch, or TensorFlow

Preferred Qualifications

• Experience developing and maintaining evaluation pipelines for ML models
• Experience deploying ML models for computer vision, NLP, robotics, or recommendation systems
• Experience supporting a small team of MLEs developing production-grade models
• Strong understanding of metrics computation and regression detection at scale