Principal Software Engineer, ML Flywheel Technical Lead jobs in United States
cer-icon
Apply on Employer Site
company-logo

Waymo · 21 hours ago

Principal Software Engineer, ML Flywheel Technical Lead

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. The Principal Software Engineer, ML Flywheel Technical Lead will be responsible for architecting and improving the machine learning self-improvement process, collaborating closely with various teams to enhance the Waymo Driver's capabilities.

Artificial Intelligence (AI)AutomotiveAutonomous VehiclesSensorTransportation
check
H1B Sponsor Likelynote

Responsibilities

Architect a path towards every autonomous mile driven by a Waymo car to be automatically incorporated into an automated data-driven self-improvement loop for the Waymo Driver
Enable a data flywheel to serve the demands of scalable pre-training, post-training targeted to relevant critical behaviors, as well as Driver simulation and validation
Enable a modeling flywheel to efficiently consume that data to reliably generate updated models that are validated and deployable with minimal human toil
Coordinate cross-functional efforts in partnership with data and ML infrastructure teams, resource planning, logging infrastructure, modeling and validation teams to accelerate the velocity, impact and leverage of driving data on the Waymo Driver
Act as the steward of data quality, by providing tooling and metrics to evaluate the impact of mining, selection and curation on the modeling pipeline performance
Articulate the strategy for incorporating diverse data sources, including third-party and synthetic data into that improvement flywheel
Drive innovation across all axes of performance and efficiency, from Driver quality, to scalability, cost, engineering velocity, model architecture and performance

Qualification

ML model developmentLarge-scale vision modelsMachine learning flywheelTechnical leadershipData quality improvementMulti-modal model developmentCommunication skillsInterpersonal skills

Required

Master's degree or PhD in Computer Science, Engineering, or a related technical field
10+ years of experience in ML model development, and you have 2+ years experience with large-scale vision, video, or multi-modal foundation model development and their integration in end-to-end models
6+ years of experience in ML-driven production systems that develops models with large-scale data, training, evaluation, and deployment
6+ years of experience in a technical leadership role leading technical teams and setting technical directions in large ML Engineering organizations
Demonstrated expertise in large-scale machine learning and its key components: pre-training, post-training, and validation. Deep understanding of both infrastructure and quality aspects
Track record of architecting and standing up a machine learning flywheel at scale for mission critical applications
Experience with driving model quality improvements through systematic data scaling and curation
Communication and interpersonal skills. Ability to inspire, influence and coordinate across functions and disciplines

Preferred

PhD in computer science
Experience in multi-modal LLM model development, and their infrastructure
Familiarity with multi-task, end-to-end models and their development challenges

Benefits

Waymo’s discretionary annual bonus program
Equity incentive plan
Generous Company benefits program

Company

Waymo is a mobility technology company that improves transportation by developing self-driving solutions for travelers and daily commuters. It is a sub-organization of Alphabet.

H1B Sponsorship

Waymo has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (231)
2024 (175)
2023 (268)
2022 (306)
2021 (298)
2020 (317)

Funding

Current Stage
Late Stage
Total Funding
$11.1B
Key Investors
Alphabet
2024-07-23Series C· $5.6B
2021-06-16Series B· $2.5B
2020-05-12Series A· $750M

Leadership Team

leader-logo
Tekedra Mawakana
Co-Chief Executive Officer
linkedin
leader-logo
Annabel Chang
Head of State Policy & Government Relations
linkedin
Company data provided by crunchbase