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.
- 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
- International Standardization Lead (Staff Researcher)London, England, United Kingdom
- Technical Program Manager, Vehicle SecurityNovi, Michigan, United States
- Senior Software Engineer, Internal ToolingMountain View, CA, US
- Machine Learning Engineer, Simulation RealismMountain View, CA, USA
- Tech Lead Manager ML OptimizationMountain View, California, United States
- Machine Learning Engineer Perception LLM/VLM (PhD, New Grad)Mountain View, CA USA; San Francisco, CA USA;
- Staff Machine Learning Engineer - Vision-Language Foundation ModelsMountain View, California, United States
- Tech Lead Manager- Onboard Software PerformanceMountain View, CA, USA