Staff ML Engineer, Perception Research

Waymo
Mountain View, California, United States
Full Time
Hybrid
Senior
USD 251,000 – 310,000
Postgraduate Degree
Technology Information Technology Telecommunications

Posted 2 days ago

Apply to this job and hundreds more automatically

Get Started
Excellent 4.7

Job summary

Waymo is seeking a Staff ML Engineer with a focus on perception research, specializing in advanced multimodal large language models and world models for 3D perception using sensor data. Responsibilities include developing scalable ML training/deployment pipelines, optimizing models, and working with distributed inference infrastructure. The role requires significant expertise in reinforcement learning, foundation models, and large-scale model training, with a focus on autonomous driving applications.

Responsibilities

Conduct comprehensive experimentation to train and deploy state-of-the-art Multimodal LLMs and World models for 3D perception. Develop and maintain scalable data pipelines and evaluation frameworks to support model development and continuous training.

Qualifications

Data Pipelines Deep Learning Distributed Training JAX, Flax, TensorFlow, PyTorch Large-scale Inference Infrastructure Machine Learning Model Optimization Perception Models Reinforcement Learning Transformers

Requires a PhD or Masters in Computer Science, Machine Learning, Robotics, or a similar field with 4+ years of industry or post-doc research experience. Proficiency in scalable model training flows and deep experience with Transformer architectures is essential.

English

Benefits

  • Health insurance
  • Dental insurance
  • Vision insurance
  • Life insurance
  • Disability insurance
  • 401(k) with company match
  • Paid time off
  • Sick time
  • Maternity leave
  • Baby bonding leave
  • Paid holidays
  • Annual bonus program
  • Equity incentive plan

Company

Waymo

Waymo

linkedin
Technology, Information and Internet
Location
Mountain View, California, United States
Website
waymo.com

Apply to this job and hundreds more automatically

Get Started
Excellent 4.7
Pete DeQuarto avatar
Pete DeQuarto

AIApply is a total game-changer. Thanks to Auto-Apply, I have been applying to jobs I would never even have come across otherwise!

Akarshak Tanwar avatar
Akarshak Tanwar

In one week I applied to over 150 jobs and locked in 2 confirmed interviews, with a support team that was quick to respond. The auto-apply feature is fascinating and a real pleasure to use!

JC
Julian Carlin

It really worked. In under 48 hours it had applied to 100 jobs on my behalf without any time or effort from me. I am looking forward to seeing how it keeps improving.