Staff Tech Lead, Machine Learning Engineer, PerceptionActive$310K

The opportunity

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…

What you'll do

  • Lead the design and development of multi-sensor, multi-task model: architectures for dense and complete scene understanding with a focus on key tasks including object detection, occupancy prediction, road understanding, and trajectory planning.

  • Architect and develop the foundational modeling libraries and software: powering model development and experimentation.

  • Design and implement scalable core model components, including sensor: encoders, sensor fusion modules, backbones, and temporal aggregation / memory blocks.

  • Adapt SOTA ML technologies to unique compute requirements, leading: optimizations for latency, throughput, and memory to operate efficiently within strict compute budgets.

  • Drive rapid prototyping by building and maintaining flexible pipelines and: development environments for fast iteration on novel model designs, enabling seamless integration of field advancements and in-depth analysis of model training dynamics and learned representations.

  • Bachelors in Computer Science or a similar discipline, or an equivalent amount of deep learning experience

What they're looking for

  • + years experience in Machine Learning and/or Computer Vision
  • Strong expertise in Python and C++ and a solid software engineering background
  • Experience with ML frameworks like PyTorch, JAX, or Tensorflow
  • MS or PhD Degree in Machine Learning, Robotics, Computer Science or a similar discipline