PitchMeAI Jobs
Waymo

Senior/Staff Machine Learning Engineer, Infrastructure

Waymo · Mountain View, CA, USA; San Francisco, CA, USA

Posted about 3 months ago

or apply directly on Waymo's site. We never take the application ourselves.

Is this posting real?

This role has been open
80 days
Waymo's roles stay open a median of 81 days
Reposted
No
Salary listed
No
0% of Waymo's roles list one
Ghost-job risk at Waymo
high
348 stale, 9 reposted of 353 open
Hiring momentum
490 roles opened in the last 90 days
↑ up vs. the prior 90 days
Last confirmed on the employer's board
2026-10-23

Measured from postings appearing on and disappearing from Waymo's own greenhouse board since 2026-08-03. Full hiring picture for Waymo.

About this role

As a Staff Machine Learning Engineer in Infrastructure at Waymo, you will design and optimize simulation environments and business logic using JAX and TensorFlow on TPU hardware. Your role will involve collaborating with modeling teams, driving technical architectures, and profiling systems to enhance performance. You will also translate business goals into technical requirements and deliverables, focusing on large-scale ML systems and their execution efficiency.

Our read on this posting2.6out of 5
benefits
2/5
freshness
1/5
career value
5/5
role clarity
5/5
pay transparency
0/5

Scored from the posting itself — how clearly the role is described, how much it says about pay and benefits, and how recently it was listed. Not a judgement of Waymo as an employer.

What you need

  • 6+ years of professional software engineering experience
  • At least 4 years focused on machine learning infrastructure
  • Direct ML programming experience on TPU and GPU hardware using JAX, PyTorch, or TensorFlow
  • Experience scaling large models using model parallelism, data parallelism, or distributed training techniques
  • Strong understanding of state-of-the-art ML models and proficiency with ML accelerator profiling tools
  • Ability to independently lead technical initiatives and build robust libraries and pipelines

Nice to have

  • Practical experience in Reinforcement Learning (RL), Sim2Real transfer, or Robotics
  • Experience with distributed ML frameworks and accelerators like GPU/TPU
  • Domain familiarity with Autonomous Driving systems, multi-agent simulations, or realistic world modeling

What you get

  • Discretionary annual bonus program
  • Equity incentive plan
  • Generous Company benefits program

Worth weighing

  • Salary range provided is broad and depends on various factors
  • Role involves advanced technical skills and may require working on complex systems
  • Focus on specific technologies like JAX and TensorFlow may limit exposure to other frameworks

Summarised from Waymo's posting. Read the full original.

Listed by Waymo on their greenhouse job board, last confirmed open on 2026-10-23. PitchMeAI is not the employer.

More roles at Waymo

All 353 open roles at Waymo →