waymo
Principal Software Engineer, ML System Architect
Mountain View, San Francisco · Principal · Full-time
Sponsorship not specified$349k-$431kDetected 12 days ago
PythonC++Machine LearningDeep LearningData EngineeringResearchLeadershipCommunicationMentoring
About the role
- The Waymo Driver powers Waymo's fully autonomous ride-hail service and can also be applied to a range of vehicle platforms and product use cases.
- The Waymo Driver has provided over ten million rider-only trips, enabled by its experience autonomously driving over 100 million miles on public roads and tens of billions in simulation across 15+ U.S. states.
- The expected base salary range for this full-time position across US locations is listed below.
Responsibilities
- Drive the consolidation of tokenization/de-tokenization strategies, data formats, input pipelines, and evaluation methodologies across all offboard Foundation Model use cases.
- Drive alignment on technical decisions with senior stakeholders across Waymo and Google Deepmind.
- Waymo's Systems Intelligence and ML team works with Research and Production teams to develop and deploy models that are core to our autonomous driving software.
- You will be the technical authority defining how Waymo builds, trains, and utilizes these large models offboard to ultimately accelerate onboard deployment and improvements.
- Demonstrated ability to design robust, scalable, and maintainable software architectures and APIs.
- Technical leadership skills, with the ability to drive strategy, influence across teams, and mentor other engineers.
Requirements
- Master's degree or PhD in Computer Science or a related field.
- Extensive experience with large-scale distributed training on TPUs/GPUs and associated challenges.
- Experience with multimodal and generative models.
- Experience with simulation systems.
Skills
- Serve as the primary technical interface between Waymo's offboard model development and Google Deepmind's core model and framework teams.
Compensation
- $349,000 - $431,000 USD
- The expected base salary range for this full-time position across US locations is listed below.
Benefits
- We are seeking a deeply experienced Principal Software Engineer to provide the overarching technical vision, architectural design, and cross-team leadership to make Waymo's foundation model systems nextgen a success.
- Establish and enforce rigorous coding standards, testing practices, and API designs to ensure long-term codebase health and developer velocity.
- 12+ years of experience in software engineering, with at least 8+ years focused on large-scale machine learning systems, deep learning frameworks, and AI infrastructure.
- Communication skills, with the ability to articulate complex technical vision and drive alignment, capable of conveying complex technical ideas clearly.
- In this hybrid role, you will report to our Director of Engineering who leads Systems Intelligence and Machine Learning.
Company info
- Waymo's AI is at the heart of this mission, and we are increasingly leveraging large-scale Foundation Models to unlock new capabilities for the Waymo Driver.
This listing is sourced directly from waymo's careers page and normalized into a canonical job model.