2022-01-05 Waymo

Senior Software Engineer, 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 ML 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 as well as our collaborators across Alphabet.

We are looking for engineers with a variety of backgrounds and are passionate about enabling ML to join our team. Current projects include:

  • Collaborating with hardware architects, software engineers, and data scientists to take full advantage of Waymo’s hardware architecture as well as innovate to enable autonomous driving.
  • Developing and maintaining a platform and ML libraries that enable a fast research to production lifecycle, including tools to support model evaluation, model debugging, hyperparameter tuning, benchmarking, and model optimization.
  • Developing Waymo’s compiler powering inference of deep nets and compute workloads with an emphasis on performance.
  • Developing benchmarking and profiling tools to optimize inference workloads and analyze performance behavior within the car.

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

  • B.Sc. in Computer Science, Mathematics, or a related field.
  • Strong C++ and/or Python programming skills.
  • 5+ years of experience developing large-scale distributed computing or data infrastructures.
  • Experience with deep learning frameworks (e.g. TensorFlow, PyTorch, and Apache MXNet).

It’s preferred if you have one or more of the following:

  • M.S. or Ph.D. degree Computer Science or related quantitative field with a specialization of machine learning.
  • Experience with building infrastructure and tools for applied machine learning, including MLOps, evaluation/validation techniques, and model performance optimization.
  • Experience with the following technologies: XLA, TVM, MLIR, LLVM, deep learning models and algorithms, and deep learning framework design.
  • Experience developing and deploying models for edge devices.
  • Experience in compiler design for CPU and off-load engines such as GPU, NPU, and TPU.

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 due to a disability, medical condition or religious belief, Waymo will provide a reasonable accommodation consistent with federal, state, and local law.

We appreciate your interest in Waymo. Waymo is an equal employment opportunity employer. Waymo’s policy is not to discriminate against any applicant or employee based on race, color, sex, religion, national origin, age, disability, military status, genetic information or any other characteristic protected by law. Waymo also prohibits harassment of applicants or employees based on any of these protected categories. Waymo will also consider for employment qualified applicants with criminal records in accordance with applicable law. Waymo also makes workplace accommodations for qualified individuals with disabilities as required by applicable law.

Apply now

Your location*
Resume/CV*

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

Cover Letter

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

If comfortable, please share your preferred gender pronouns*

Waymo is committed to maintaining the same safe and inclusive environment for candidates as for our team members. Addressing you by the correct pronoun is an important part of this commitment.

Work Authorization*
How did you hear about this opportunity?
Please review and acknowledge our Candidate Privacy Policy linked below:*
Are you a current or former Alphabet employee (including Google)?*

Demographic Information (completion voluntary)

You are being given the opportunity to provide the following information in order to help us track and measure the aggregate demographics of our applicants in furtherance of our Equity, Inclusion, and Diversity efforts.  If you decide to provide demographic information, it will be recorded and maintained in a confidential file.  It will not be considered in the hiring process or as a basis for employment decisions.  Answering these questions is entirely voluntary. If you are offered and accept employment with Waymo, the information collected during the application and recruitment process will become part of your employment record.

Gender
Race
Are you a veteran or reservist?
Do you identify as a member of the LGBTQ+ (e.g., lesbian, gay, bisexual, queer, asexual) community?
Sorry, an error occurred. Please verify all fields have been filled and try again.
Your application has been successful.
check_circle Thank You
You should receive an e-mail within the next few minutes.