Zoox logo

Senior/Staff Software Engineer - Machine Learning & System Optimization

Zoox

As a Machine Learning and System Optimization Engineer, you will orchestrate and allocate overall system capacity to various core perception models running on-bot, as well as drive large initiatives that allow for more efficient inference…

Apply on Zoox’s careers page
Zoox
Foster City, United States
Hybrid
Full-time
Staff / Principal
Robot Learning & AI
Autonomy Software
$254K–$350K
30 June 2026

About the role

The Perception team is pioneering the development of a multi-modality foundation model to drive the next generation of autonomous system intelligence. As a Machine Learning and System Optimization Engineer, you will orchestrate and allocate overall system capacity to various core perception models running on-bot, as well as drive large initiatives that allow for more efficient inference by sharing various parts of the perception stack with one another. You will focus on bringing highly efficient, production-ready large-scale models to our on-vehicle stack. We are looking for experts with hands-on experience compressing, accelerating, and deploying complex models, including LLMs, VLMs, or foundation models, for power- and thermal-constrained vehicle SoCs. In addition, you will optimize ML models, write custom CUDA kernels, and build highly concurrent inference code to ensure real-time, deterministic execution on edge devices.

More open roles at Zoox

All 241 roles at Zoox

Other Robot Learning & AI roles in robotics

All Robot Learning & AI roles