2020-03-30 Waymo
Software Engineer, Machine Learning Infrastructure (Training Infrastructure)
Mountain View, California, United States

Waymo is the self-driving technology company with a mission to make it safe and easy for people and things to move around. Building on software and sensor technology developed at Google, Waymo is now launching the world’s first fully self-driving transportation service that will take members of the public from A to B at the touch of a button.

The Machine Learning Infrastructure team at Waymo builds the industrial scale machine learning platform used by ML practitioners developing software for our self-driving car. We provide essential tools and frameworks to support the entire lifecycle of machine learning at Waymo, from data processing, large scale training and evaluation frameworks to efficient neural net inference runtimes for onboard execution and simulation. We work hand in hand with machine learning experts in all parts of the company including perception, behavior prediction, planning, mapping, hardware, and simulation. 

Join the Training Infrastructure group within Machine Learning Infrastructure, and help us make training and evaluating ML models at Waymo easier, faster, and better!  We develop and maintain a set of frameworks and tools on top of Tensorflow that address many of the pain points experienced by ML practitioners: training fast and at scale, discovering optimal hyper-parameters, automatically retraining nets on a schedule, computing reliable and noiseless metrics on validation sets, and validating newly trained nets when deployed into the full onboard software stack. We collaborate with teams at Google and DeepMind to leverage their research and infrastructure, as well as develop our own in-house techniques.

Basic Qualifications:

  • Solid Python or C++ skills
  • Passionate about infrastructure work, building libraries, tools, and pipelines for machine-learners
  • Experience with Tensorflow, Keras etc.
  • Experience contributing to the architecture and design (architecture, design patterns, reliability and scaling) of new and current systems
  • BS, MS or PhD in CS, Math, or equivalent real-world experience

Bonus Qualifications in one of the following, depending on the project:

  • Knowledge or experience in training models with TPU
  • Experience building and architecting large-scale, production quality backend systems, especially in applied machine learning or data pipeline
  • Front-end experience in designing and building user interfaces that help machine learning developers explore datasets, track experiments, visualize metrics and monitor deployed models
  • Knowledge and experience with machine learning algorithms
  • Experience with multi-threaded and stream-based programming models
  • Experience with high performance computing or data mining
At Waymo, we don’t just accept difference - we celebrate it, we support it, and we thrive on it for the benefit of our employees, our products and our community. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or Veteran status. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. If you have a disability or special need that requires accommodation, please let us know.
Apply now
Your location
Cover Letter
Are you currently or were you previously an Alphabet employee?*
Do you have internal references at Waymo and/or another Alphabet company?*
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.