Staff Software Engineer/Tech Lead - Onboard Model Consolidation

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

• Lead the software architecture and infrastructure development for onboard model consolidation across many onboard subsystems
• Design and build an onboard framework to support flexible and efficient model integration and inference in onboard and simulation environments
• Spearhead the alignment and unification of model’s data generation subsystem between model training data preparation and model inference
• Develop Simulation solutions to enable robust testing and efficient evaluation/validation
• Embed deeply with model teams to drive the productionization of unified models
• Establish new software development best practices and technologies to accelerate model development and releases

Requirements

• BS/MS in Computer Science, EE, Robotics, Physics, Math, or related field
• 8+ years of professional software development experience on large scale products
• Extensive experience designing and building large-scale, high-performance C++ software architecture
• Proven track record of leading complex, cross-functional engineering projects as a technical lead
• Deep understanding of ML deployment, inference frameworks, and associated developer tooling
• Strong communication and collaboration skills

Preferred Qualifications

• Master's or PhD in Computer Science, Machine Learning, Robotics, or a related field
• Experience with data pipelines and feature extraction for machine learning models
• Prior experience in the autonomous driving industry (onboard compute, perception, or planning)
• Experience with hardware accelerators such as TPUs or GPUs and ML model inference in edge environments