ML Training Performance Engineer: Scalable Foundation Models

Zoox is building autonomous robotaxis and scaling ML training for VLMs, VLA, and foundation models. The ML Platform team enables innovations across distributed training, mixed-precision, and RL-driven improvements.

You will collaborate with autonomy teams to push training efficiency and throughput while integrating with Zoox’s hardware and software stack. The role emphasizes scaling training workloads, profiling bottlenecks, and developing custom kernels.

Consider building your career as a ML Training Performance Engineer:
Scalable Foundation Models at Zoox.

We are looking to fill the ML Training Performance Engineer:
Scalable Foundation Models position at Zoox in Foster City, CA, United States.

The ML Training Performance Engineer:
Scalable Foundation Models role at Zoox is now open for applications in Foster City, CA, United States.

Join us at Zoox as our next ML Training Performance Engineer:
Scalable Foundation Models in Foster City, CA, United States.