2022-08-08 Waymo

Software Engineer, ML System Performance, Machine Learning Platform

Mountain View, California, United States
San Francisco, California, United States

Waymo is an autonomous driving technology company with a mission to make it safe and easy for people and things to get where they’re going. Since our start as the Google Self-Driving Car Project in 2009, Waymo has been focused on building the World’s Most Experienced Driver in hopes of improving the world's access to mobility while saving thousands of lives now lost to traffic crashes. Our Waymo Driver powers Waymo One, our fully autonomous ride-hailing service, as well as Waymo Via, our trucking and local delivery service. To date, Waymo has driven over 20 million miles autonomously on public roads across 25 U.S. cities and conducted over 20 billion miles of simulation testing.

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. 

The Machine Learning Platform team at Waymo provides a set of tools and technologies to support and automate the lifecycle of the machine learning workflow, including feature and experiment management, model development, debugging & evaluation, deployment, and monitoring. These efforts have resulted in making machine learning more accessible to teams at Waymo, including Perception, Behavior Prediction, Planner, Routing, Maps and Research, ensuring greater degrees of consistency and repeatability, and addressing the “last mile” of getting models into production and managing them once they are in place. We work hand in hand with machine learning experts in all parts of the company and our collaborators across Alphabet.

We are looking for an individual contributor (IC) with a strong background in ML deployment toolchain, ML accelerator (GPU/TPU) profiling and application, and remote procedure call (RPC) service management and optimization. Non-exhaustive examples of our work include:

  • Develop scalable and performant inference service in both Google internal data centers and Google cloud 
  • Build a comprehensive and user friendly toolchain for scalable and flexible ML deployment workflow
  • Profile ML performance on accelerators (e.g. GPU and TPU) at both model level and system level, identify performance bottlenecks and optimization opportunities 
  • Programming languages: C++, Python, Golang

At a minimum, we'd like you to have:

  • B.Sc in Computer Science, EE or a related field
  • Strong C++ programming skills
  • Strong problem solving skills
  • 2+ years of performance analysis and optimization experience
  • Experience with multi-core systems and multi-thread interactions
  • Experience with CPU, GPU, or computer system microarchitecture

It's preferred if you have:

  • M.Sc or Ph.D in Computer Science, EE or a related field
  • C++/Python/Golang programming skills
  • Embedded / real time / robotics systems
  • Experience working with ML workload on edge device
  • ML Inference service building and management
  • General knowledge and experiences about RPC services

Note: The following is only applicable for roles hired in the US. As we keep the safety of our employees and our communities top of mind, COVID-19 vaccinations continue to be a critical prevention measure to help end the pandemic.  Because of this, Waymo requires all US-based employees who perform work onsite for Waymo to be fully vaccinated against COVID-19. If you are unable to be vaccinated, Waymo will provide a reasonable accommodation consistent with applicable law.

We appreciate your interest in Waymo. Waymo is an equal employment opportunity employer, committed to maintaining a supportive and inclusive workplace for all employees. 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, 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 ensuring equal opportunity for qualified individuals with disabilities. If you are an individual with a disability and require an accommodation to participate in the application or interview process, 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.)

Apply now

Resume/CV

.pdf, .doc, or .docx only. Attachments must not exceed 10MB.

Cover Letter

.pdf, .doc, or .docx only. Attachments must not exceed 10MB.

Sorry, an error occurred. Please verify all fields have been filled and try again.
By submitting my application, I certify that the information provided (including in my resume/CV) is true and correct. I understand that any misrepresentations or false statements made in connection with the recruitment process may be grounds for disqualification from further consideration for employment or, if employed, separation of employment.
Your application has been successful.
check_circle Thank You
You should receive an e-mail within the next few minutes.