Senior/Staff Machine Learning Engineer, Infrastructure
Waymo · Mountain View, CA, USA; San Francisco
Don’t apply blind. See how your CV matches Engineer first — free, in 30 seconds.
You will leave NewLuxJob. We do not receive or handle applications.
- Company
- Waymo
- Location
- Mountain View, CA, USA; San Francisco
- Salary
- $251,000 — $310,000
- Posted
- December 8, 2025
About this job
Join our ML Infrastructure engineering team advancing state-of-the-art ultra-realistic multi-agent simulations using foundation models. In this role, you will work at the intersection of ML infrastructure, foundation models, and simulation engineering, with a specific focus on writing high-performance business and simulation logic in JAX/TensorFlow running directly on TPUs to power realistic environments for Reinforcement Learning (RL). What You'll Do Design, build, and optimize realistic simulation environments and business logic running on TPUs using JAX and TensorFlow. Implement and optimize large-scale model and data parallelism strategies for training and running foundation models on TPU hardware.…
This is a short summary.
Want to know if you're a fit? Check your CV against this role — free, in 30 seconds.
Most-requested skills in United States
Based on 82 United States vacancies that list requirements, these are the skills employers ask for most often.
- Project Management49% of postings
- CAD26% of postings
- Quality Assurance9% of postings
- PLC7% of postings
- Lean7% of postings
- FEM/FEA7% of postings
- SolidWorks6% of postings
- MATLAB5% of postings
Similar jobs
See if your CV fits this job
Paste your CV for an instant match score against this role — and get a tailored cover letter in one click.
- Instant match score for this role
- Tailored cover letter in one click
- Free — no credit card