All jobs

Senior/Staff Software Engineer - Machine Learning & System Optimization

Zoox4h ago
United StatesOnsiteFull-timeSenior Level5+ yrs exp
H-1B verified · 2310 LCAs

Top focus

Software EngineerSenior Ml EngineerSenior Software EngineerMl EngineerSoftware Engineer Ii

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.

Required skills

Machine LearningSystem OptimizationCUDA
Posted on JobRush — the end-to-end AI job-search platform.