Senior ML Lead - Perception & Multi-Sensor Vision (Equity)

Waymo

Mountain View (CA)

On-site

USD 251,000 - 310,000

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Job summary

Waymo is seeking a senior machine learning and computer vision leader to drive the design of multi-sensor, multi-task models for dense scene understanding such as object detection and trajectory planning. You will architect core modeling libraries and scalable components, and optimize for latency, throughput, and memory to meet compute budgets.

Ideal candidates have 7+ years in ML/CV, strong Python/C++, and experience with PyTorch/JAX/TF.

Qualifications

Bachelor's in Computer Science or equivalent deep learning experience.
7+ years of experience in Machine Learning and/or Computer Vision.
Strong Python and C++ with a solid software engineering background.
Experience with ML frameworks like PyTorch, JAX, or TensorFlow.

Responsibilities

Lead design and development of multi-sensor, multi-task model architectures for dense and complete scene understanding (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 compute requirements, leading optimizations for latency, throughput, and memory to operate within strict budgets.
Drive rapid prototyping by building and maintaining flexible pipelines and development environments for fast iteration on novel model designs and analysis of training dynamics.