jobdescription_image_2x.png

Principal Software Engineer, ML Flywheel Technical Lead

Mountain View, California, United States. San Francisco, California, United States Full-Time Software Engineering 4515

Apply now

Principal Software Engineer, ML Flywheel Technical Lead

  • 4515
  • On Site
  • Mountain View, California
  • San Francisco, California
  • Software Engineering
  • Full-Time
View favorites

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver—The World's Most Experienced Driver™—to improve access to mobility while saving thousands of lives now lost to traffic crashes. 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.

Software Engineering builds the brains of Waymo's fully autonomous driving technology. Our software allows the Waymo Driver to perceive the world around it, make the right decision for every situation, and deliver people safely to their destinations. We think deeply and solve complex technical challenges in areas like robotics, perception, decision-making and deep learning, while collaborating with hardware and systems engineers. If you’re a software engineer or researcher who’s curious and passionate about Level 4 autonomous driving, we'd like to meet you. 

Every mile driven by a Waymo car is a unique and valuable piece of data that can be leveraged into improving our machine learning models, and ultimately the safety and capabilities of the Waymo Driver. Charting the path from collection of a piece of data, to curation, (auto-)labeling, model training and evaluation, all the way to model deployment and monitoring is a process of continuous scaling and quality improvement of the entire ML lifecycle. The Area Technical Lead for the Waymo machine learning flywheel is responsible for architecting, connecting, automating and improving the entire span of this self-improvement process in close collaboration with Waymo’s infrastructure, modeling and evaluation teams. They are directly accountable to Waymo’s leadership for the organization’s ML data strategy and its impact on Driver quality.

You will report directly to our Distinguished Engineer, Foundation Models.

You Will:

  • 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.

You Have:

  • 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.

We Prefer:

  • 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.

The expected base salary range for this full-time position across US locations is listed below. Actual starting pay will be based on job-related factors, including exact work location, experience, relevant training and education, and skill level. Your recruiter can share more about the specific salary range for the role location or, if the role can be performed remote, the specific salary range for your preferred location, during the hiring process. 

Waymo employees are also eligible to participate in Waymo’s discretionary annual bonus program, equity incentive plan, and generous Company benefits program, subject to eligibility requirements. 

Salary Range
$332,000—$421,000 USD

We appreciate your interest in Waymo. Waymo is proud to be an equal opportunity employer, committed to creating a culture of belonging and maintaining a supportive workplace for all employees. We welcome applicants of all backgrounds, and employment decisions are based on a candidate’s qualifications, experience, and alignment with job requirements and business needs. Waymo does not discriminate against, and prohibits harassment of, any applicant or employee based on race, color, sex, sexual orientation, gender identity, religion, national origin, age, disability, military status, family status, pregnancy, genetic information or any other basis protected by applicable law. Waymo will also consider for employment qualified applicants with criminal records in accordance with applicable law. Waymo is committed to making sure our hiring process is accessible for all candidates. If you need assistance applying for a role or participating in the interview process due to a disability, please let the recruiting team know or email waymo-candidatesupport@google.com. (This email address is intended to be used only for requesting accommodations as part of the application process. Other inquiries will not receive a response.)

Some of our Benefits

Health and wellness

Our people are at the heart of everything we do. At Waymo, you can enjoy top-notch medical, dental and vision insurance, mental wellness support, gym membership, and special wellness programs.

Financial wellness

Your financial peace of mind is important to us. At Waymo, we offer competitive compensation, bonus opportunities, equity, employees provident fund, and lots of other perks and employee discounts.

Flexibility and time off

Take the time you need to relax and recharge. Enjoy the flexibility to work from another location for four weeks per year. We support an on-site or hybrid work model and offer remote working opportunities, paid time off, bereavement, sick, and parental leave. 

Supporting families

When it comes to growing your family or caring for your loved ones, you have our full support. Enhanced leave options include 26 weeks of paid leave for birthing parents and 18 weeks of paid leave for non-birthing parents.

Community and personal development

At Waymo, you'll find a range of opportunities to grow, connect, and give back. We offer education reimbursement, personal and professional development, mentorship, and other ways to connect through Employee Resource Groups (ERGs), other internal groups, and even time off to volunteer.

Cool perks

Access to Google offices, cafes, wellness centers, massages, and so much more. To support your wellbeing at home, you can enjoy at-home fitness and cooking classes, and more.

Ready to Apply?