Machine Learning Engineer, Waymo

Training

$158-200k

+Equity and Bonus

AWS
GCP
Python
Tensorflow
C++
Azure
PyTorch
Junior and Mid level
San Francisco Bay Area

Office located in Mountain View, CA

Waymo

Autonomous driving technology

Job no longer available

Waymo

Autonomous driving technology

1001+ employees

B2CTransportAutomation

Job no longer available

$158-200k

+Equity and Bonus

AWS
GCP
Python
Tensorflow
C++
Azure
PyTorch
Junior and Mid level
San Francisco Bay Area

Office located in Mountain View, CA

1001+ employees

B2CTransportAutomation

Company mission

To make it simpler and safer for people to move things and themselves around.

Role

Who you are

  • Bachelor's degree in Computer Science, Engineering, or related field, or 2+ years equivalent experience
  • Experience with distributed systems principles and experience building distributed systems for production environments
  • Solid Python or C++ skills
  • Prior experience with Machine Learning frameworks (e.g., TensorFlow, PyTorch) and distributed training algorithms
  • Debug complex distributed systems issues
  • Experience communicating updates and resolutions to customers and other partners

Desirable

  • Practical familiarity using ML accelerator profiling tools to uncover performance bottlenecks
  • Familiarity with cloud computing platforms (e.g., AWS, Azure, GCP) and experience deploying and managing distributed systems in cloud environments
  • Knowledge of optimization and deep learning algorithms

What the job involves

  • The Waymo ML Infrastructure team works with Research and Production teams to develop models in Perception and Planning that are core to our autonomous driving software
  • We ensure our partners by offering the best solutions for the entire model development lifecycle
  • These solutions are developed in close collaboration with teams at Google. They are geared towards both scaling models and solving problems unique to ML for autonomous driving
  • Report into the TLM of ML Training
  • Develop the infrastructure components necessary for distributed training, including job scheduling, resource management, data distribution, and model synchronization
  • Implement automation solutions for provisioning, deployment, monitoring, and scaling of distributed training infrastructure to improve operations and reliability
  • Monitor system health, diagnose and troubleshoot issues, and perform routine maintenance tasks to ensure the reliability of the distributed training infrastructure
  • Identify performance bottlenecks and optimization opportunities
  • Improve the developer experience and performance of our scalable ML framework

Share this job

View 87 more jobs at Waymo

Insights

Led by a woman
Top investors

-1% employee growth in 12 months

Company

Funding (last 2 of 4 rounds)

Aug 2024

$5.6bn

SERIES C

Jun 2021

$2.5bn

GROWTH EQUITY VC

Total funding: $11.1bn

Our take

Waymo is an autonomous driving technology company that began as the Google self-driving car project in 2009. After initial development, it secured permits to deploy autonomous ride-hailing services in cities including Phoenix and San Francisco, but has recently hit a major bump in the road with safety concerns that resulted in a recall of over 400 self-driving cars.

With the autonomous driving industry already besieged by setbacks and public distrust, this development does Waymo no favours. That said, as an industry leader there remains the possibility it will see out the storm, but massive attention will need to be given to rebuilding trust with consumers if this is to be the case.

Steph headshot

Steph

Company Specialist at Welcome to the Jungle