zoox Verified 10h ago
Senior/Staff Software Engineer - Machine Learning & System Optimization
Seattle, WAAll locationsSeattle, WAFoster City, CABoston, MA Hybrid
$254,000–$350,000 a yearPay$254,000–$350,000
TypeFull-time
Work settingHybrid
Verified listing
JobFig found this opening at its original source and checks that it remains available.
About the role
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.
What you'll bring
- 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.
- Deep experience in system and performance optimization in CPU/GPU systems designed for low latency or high throughput.
- Deep expertise in working with real-time systems & required constraints such as processing latency, memory utilization, and memory bandwidth pressure.
- Deep expertise in model quantization (PTQ, QAT) and mixed-precision inference frameworks (INT8, FP8, FP4, BF16/FP16).
- Proficiency in low-level programming for AI accelerators, specifically developing and optimizing custom ML OPs and TensorRT Plugins with efficient CUDA kernel implementations.
- Production-level C++ (14/17/20) and Python programming skills, with experience developing concurrent, memory-safe, real-time inference code for edge devices.
- You do not need to match every listed expectation to apply for this position.
- Prior experience in high-performance robotics applications such as AV/drones/robots.
- Familiarity with SOTA autonomous driving perception algorithms (temporal 3D object detection, BEV, 3D Occupancy Networks) and multi-modal sensor processing (Vision, LiDAR, Radar).
- Experience with end-to-end autonomous driving paradigms (VLM/VLA models, Foundation models) and edge deployment technologies (e.g., TensorRT-LLM).
Benefits
Disability InsuranceLife InsuranceMedical InsurancePaid Time Off
Locations
Seattle, WAFoster City, CABoston, MA