Staff Machine Learning Engineer, Infrastructure
Waymo- Location
- Mountain View, California, Usa; San Francisco, Ca, Usa
- Work mode
- Remote
- Type
- full time
- Listed
Applications go directly to Waymo. Ramai takes no fee for placement and no money for ranking.
[ About this role ]
Senior/Staff Machine Learning Engineer, Infrastructure at Waymo. This role focuses on advancing state-of-the-art ultra-realistic multi-agent simulations using foundation models. Responsibilities include designing, building, and optimizing simulation environments and business logic on TPUs using JAX/TensorFlow, and integrating foundation models into simulation pipelines.
Skills and focus areas
- machine learning infrastructure
- large-scale ml systems
- tpu programming
- jax
- tensorflow
- simulation engineering
- foundation models
- reinforcement learning
- model parallelism
- data parallelism
- system design
- performance optimization
Summary written by Ramai from the original posting. Full details and requirements are on Waymo’s careers page.
[ More at Waymo ]
See all 372 open roles at Waymo →[ Similar roles at other startups ]
- Senior Software Engineer, Agent PlatformShopMy
- Staff Security EngineerEight
- Senior Embedded Software EngineerEight Sleep
[ Not quite right? ]
Tell the assistant what you’re after — stage, location, stack, visa sponsorship — and it searches every open role for you. A few questions are free, no account needed.