Software Engineer, ML Training Optimization

Zoox is hiring a Software Engineer, ML Training Optimization in Foster City, United States. It pays $177k-$270k a year and Level rates it ; you can apply on Level.

Level's read on this job

On Level's scale, this software engineer role is Builds AI ●●●●: the job is building AI systems.
The listed pay starts 5% above the typical software engineer posting on Level.
Level lists 237 other open roles at Zoox.
The role was posted within the past week.

AI in this 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 for in- and off-vehicle ML use cases. You will work alongside a team of strong software engineers and act as a force multiplier for our internal customers. This team has many growth opportunities as we expand our robotaxi deployments and venture into new ML domains. If you want to learn more about our stack behind autonomous driving, please look here. If you want to learn more about our ML Infrastructure, here is one of our past talks at re:Invent.

How we rate this

Software Engineer, ML Training Optimization at Zoox rates 94 out of 100 for how much of the daily work is AI. That makes it Builds AI (AI Level 4 of 4). The level is about AI in the job, not seniority.

Classification

Builds AI. The job is building AI systems.

●●●● Builds AI80 to 100
●●●○ Works on AI60 to 79
●●○○ Uses AI40 to 59
●○○○ Little AI0 to 39

Levels come from how often the tools, models and workflows of the role are named in the posting itself. Open the description and count.

Key facts

Software Engineer, ML Training Optimization is an open job at Zoox.
Level rates this role AI Level 4 of 4: Builds AI.
The listed salary is $177k-$270k a year.
The role is based in Foster City, CA.
The job was posted on October 16, 2026.
Applications are handled by Zoox. Apply on Zoox

Prepare for this job

Built from this posting and from similar jobs on Level.

Also common in software engineer jobs at this level

Python LLM AWS Machine learning C++AI agents

Pay

This job starts at: $177k
Typical software engineer job: $168k

This posting starts 5% above the typical software engineer job. See software engineer jobs

Want an expert to read your CV for this job?

Free. Send your CV and the role you want next. We reply by email within 2 to 4 business days.

Get a free CV review Check your resume with the ATS resume checker

Get new software engineer jobs (Builds AI ●●●●) by email

One email a week with the new software engineer jobs (Builds AI ●●●●), each rated for how much AI is in the work. No recruiter spam, unsubscribe in one click.

Free. One email a week. Unsubscribe in one click.