Senior Staff/Principal Software Engineer - GPU System Software, Onboard Execution (CUDA)

Location: Hybrid in Foster City, CA; Seattle, WA; or Boston, MA, United States

Company

Zoox is building a purpose-built robotaxi and operating it on public roads with riders.

What you will do

• Architect how perception, prediction, and planning workloads share onboard GPUs, including scheduling, multi-tenancy, isolation, and deterministic latency under contention.
• Own the multi-year roadmap for reducing compute per vehicle by setting latency and utilization budgets and delivering improvements across models, runtimes, middleware, and hardware.
• Guide next-generation GPU and SoC selection, topology, and memory budgets using workload models and performance projections.
• Lead performance bring-up for new compute platforms and set direction for the inference stack, including TensorRT, CUDA, custom kernels, and quantization.
• Partner with ML, autonomy, middleware, and hardware teams to design models and systems for their target hardware.
• Lead and mentor senior engineers, align directors and VPs through design documents, and make trade-offs between performance, safety, and schedule.

Requirements

• BS in computer science, computer engineering, or a related field, plus 10+ years of systems or performance engineering experience.
• Experience leading GPU or accelerator architecture across teams, or building and leading a performance engineering team.
• Deep expertise in the CUDA platform, including its programming model, runtime, and driver, and NVIDIA GPU microarchitecture from Ampere through Blackwell, including Orin and Thor.
• A proven record of setting technical direction and delivering large, measurable end-to-end improvements in latency, utilization, or cost.
• Working knowledge of ML inference optimization, including TensorRT, XLA, Triton, quantization, and fusion.
• Expert C++ development on Linux.
• Ability to align ML, systems, hardware, and leadership stakeholders around a shared technical plan.

Nice to have

• Real-time or embedded GPU experience in autonomy, robotics, AR/VR, games, or HPC.
• Production GPU multi-tenancy experience with MPS, MIG, or priority streams.
• Experience with new GPU architecture or platform bring-up, benchmarking methodology, or workload modeling and projections.

Culture & Benefits

• Work within the Software Performance organization, which owns end-to-end performance of the onboard software stack.
• Collaborate with Perception, Prediction, Planning, ML Infrastructure, and Hardware teams.
• Contribute to safety-critical, real-time compute systems operating on public roads.
• Zoox is committed to building a team with diverse backgrounds, experiences, perspectives, and skills.