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
- 102 days Waymo's roles stay open a median of 103 days
- Reposted
- No
- Salary listed
- No 0% of Waymo's roles list one
- Ghost-job risk at Waymo
- high 353 stale, 9 reposted of 353 open
- Hiring momentum
- 66 roles opened in the last 90 days ↓ down vs. the prior 90 days
- Last confirmed on the employer's board
- 2026-11-14
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-11-14. PitchMeAI is not the employer.
More roles at Waymo
- Staff Software Quality Safety Operations SpecialistMountain View, CA, United States; Remote, United States
- Senior Staff Technical Lead Manager, In-Car ExperienceMountain View, CA, USA
- Escalation Manager, Event Response Team; エスカレーション マネージャー、イベント対応チームTokyo, Japan
- Tech Lead Manager, Data EngineerMountain View, California, USA
- Senior/Staff ML Engineer, 3D/4D World Modeling, SimulationMountain View, CA, USA
- Product Manager, GrowthMountain View, CA, USA; San Francisco, CA, USA
- Structured Testing Vendor Engineering LeadMountain View, CA, USA
- Senior Data Science Manager, Driving QualityMountain View, California, USA