Staff Software Engineer / Tech Lead (Onboard Model Consolidation)

• The Architecture & ML Integration (AMI) team sits within SIML and focuses on horizontal, cross-cutting challenges spanning Waymo’s end-to-end system
• Our mission is to enable rapid evolution towards an ML-first architecture that is modular, flexible, and easy to experiment with
• We do this by improving onboard software architecture, system debuggability, and developer experience through better API designs and comprehensive pipelines
• The AMI team drives the convergence of Planner and Perception systems and tackles critical shared infrastructure components
• 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 Waymo’s 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 to enable good developer experience for consolidated model development
• Embedded 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

Benefits

• Medical, dental, and vision insurance for employees and dependents
• Employee assistance programs focused on mental health
• Personalized workplace adjustments for diverse needs and abilities, including physical, mental, and neurodivergent considerations
• Access to mental health apps
• Onsite wellness centers
• Counseling services
• Second medical opinion for you and your loved ones
• Support programs including menopause benefit
• Medical advocacy program for transgender employees
• Competitive compensation
• Regular bonus and equity performance grant opportunities
• Generous 401(k) and regional retirement plans
• Annual cross-company compensation review and pay equity analysis
• 1-on-1 financial coaching
• Fertility and growing family assistance
• Parental leave and baby bonding leave
• Backup childcare
• Survivor income benefit
• Elder care and support
• Caregiver leave
• Paid time off, including vacation, bereavement, sick leave, parental leave, disability, and holidays
• Jury duty leave
• Military leave
• Hybrid work model with remote work opportunities also available
• Educational reimbursement
• Peer learning and coaching platform
• Donation matching programs
• Volunteer hours
• Employee resource groups
• Internal community groups and local culture clubs
• Inspiring spaces to work, recharge, and collaborate with fellow Waymonauts
• On-site meals and snacks
• Fitness centers, massage programs, and ergonomic support
• On-demand fitness, wellbeing, and cooking classes
• Commuter benefits

• Have 8+ years of professional software development experience on large scale products
• Deep understanding of ML deployment, inference frameworks, and associated developer tooling
• Proven track record of leading complex, cross-functional engineering projects as a technical lead
• Good communication and collaboration skills working with cross-org partner teams
• BS/MS in Comp Sci, EE, Robotics, Physics, Math, or related field (or equivalent experience)
• Extensive experience designing and building large-scale, high-performance C++ software architecture for mission-critical systems or ML inference
• Experience with data pipelines and feature extraction for machine learning models
• Master’s or PhD in Computer Science, Machine Learning, Robotics, or a related field
• Prior experience in the autonomous driving industry, particularly focusing on onboard compute, perception, or planning systems
• Experience with hardware accelerators (e.g., TPUs, GPUs) and ML models inference in edge/onboard environments