Senior Staff Software Engineer, TLM

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

• Architect a modular, extensible ML framework capable of rapidly absorbing complex signal requirements
• Define the multi-year roadmap for signal quality, coverage, and reliability
• Oversee a pipeline processing millions of segments with a requirement for >99% cache hit rate
• Optimize high-fidelity ML inference costs across massive, petabyte-scale datasets
• Act as the primary interface for Application Owners across the company to translate simulation needs into technical requirements
• Lead a high-performing team of ML and Backend Engineers

Requirements

• Master’s or PhD in Computer Science, Machine Learning, Robotics, or a related field
• Experience managing and technically leading a team of 10+ Engineers or Research Scientists
• Proven track record of navigating complex organizations and aligning diverse customer requirements into a unified roadmap
• Hands-on experience leading ICs daily, including deep knowledge of codebases, eval/data pipelines, and model architectures
• Deep understanding of Foundation Models (LLMs, VLMs, or MMLMs) regarding post-training, fine-tuning, or inference-time scaling
• Expert-level skills in Python, Numpy, and PyTorch or JAX

Preferred Qualifications

• Experience building and scaling distributed systems or data pipelines at a petabyte scale
• Specific experience in Autonomous Driving planning, perception, or simulation research
• Proficiency in C++ and experience with production-grade ML deployment
• Strong publication record at top-tier conferences (CVPR, ICRA, NeurIPS) or leadership in major open-source projects