Waymo

Staff Tech Lead, Machine Learning Engineer, Perception

Waymo
Apply
1 day ago
Mountain View, CA, USA or San Francisco, CA, USAStaff+
H1B Sponsor

Base Salary

$251k - $310k/yr

Responsibilities

  • Lead the design and development of multi-sensor, multi-task model architectures for dense scene understanding.
  • Architect and develop foundational modeling libraries and software for model development and experimentation.
  • Implement sensor encoders, sensor fusion modules, backbones, and temporal aggregation or memory blocks.
  • Optimize machine learning models for latency, throughput, memory, and strict compute budgets.
  • Build and maintain flexible prototyping pipelines and development environments for rapid model iteration and analysis of training dynamics and learned representations.
  • Collaborate with downstream teams on optimization and integration into the Waymo Driver.

Requirements

  • Bachelor’s degree in Computer Science or a similar discipline, or an equivalent amount of deep learning experience.
  • At least 7 years of experience in machine learning and/or computer vision.
  • Strong expertise in Python and C++ with a solid software engineering background.
  • Experience with machine learning frameworks such as PyTorch, JAX, or TensorFlow.

Benefits

  • Hybrid work arrangement.
  • Eligibility for a discretionary annual bonus program, equity incentive plan, and company benefits program, subject to eligibility requirements.
  • Full-time position.

Tech Stack

Categories

Waymo

About Waymo

1,001-5,000 employees

On the journey to be the world's most trusted driver. With the Waymo Driver, we can improve mobility while saving thousands of lives. Download the Waymo One app and ride today. Waymo reaches out to candidates from official channels only (e.g. directly from @waymo.com email addresses, or through our recruiters or sourcers who are noted as such on LinkedIn). We do not contact candidates about career opportunities through instant messaging apps like Telegram, email addresses from domains other than waymo.com (such as Gmail addresses), direct messages on Twitter, Facebook, and Instagram, or text messages. Visit waymo.com to check out our official job listings. Need support? Send us a message – we're here to help.

Contact me