Software Engineer, ML Training Optimization

Zoox·Foster City, CA·Posted 16 October 2026

Apply with a tailored CV View on Zoox's site

About the role

Zoox is on a mission to reimagine transportation and ground-up build autonomous robotaxis that are safe, reliable, clean, and enjoyable for everyone. We are still in the early stages of deploying our robotaxis on public roads, and it is a great time to join Zoox and have a significant impact in executing this mission. The ML Platform team at Zoox plays a crucial role in enabling innovations in large-scale Foundation models, VLMs, and VLAs to make autonomous driving as seamless as possible. The Opportunity Are you excited to drive our ML Training Optimization initiatives and make the models that power autonomous driving trains faster, cheaper, and at greater scale? You will work with SOTA accelerators and cutting-edge techniques in distributed training, parallelism strategies, mixed-precision training and RL, especially for VLA and VLM models that are multi-modal — partnering closely with all the Autonomy teams within Zoox (Perception, Prediction, Planner, Simulation, Collision Avoidance) and pushing the boundaries of how ML training is practiced at Zoox. We build and operate the base layer of ML tools, model development, and training systems that our applied research teams use…

This is a summary of the advert. The full description and the application are on Zoox's careers site.

Apply faster with Prepper

A version of your CV tailored to this Software Engineer, ML Training Optimization role.
A cover letter written from your CV and this advert.
Prepper can fill in Zoox's application form and send it once you approve.
The application saved in your tracker, with reminders to follow up.

Start free

More roles at Zoox

Prepper is independent and not affiliated with Zoox. Roles are listed from the employer's public careers system and removed when they close.