Software Engineer - C++ GPU Performance (Robotics)

TL;DR

Software Engineer - C++ GPU Performance (Robotics): Instrument, monitor, analyze, and optimize GPU-based algorithms for self-driving systems with an accent on performance monitoring, hotspot identification, and kernel optimization. Focus on developing real-time instrumentation tools, porting serial algorithms to GPU, and designing middleware frameworks for efficient code development.

Location: Hybrid in Foster City, CA / Boston, MA / San Diego, CA / Seattle, WA

Salary: $168,000 - $239,000 a year

Company

Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem at the intersection of robotics, machine learning, and design.

What you will do

• Build real-time instrumentation for performance monitoring (CPU, GPU, latency, memory) and offline benchmarking frameworks to evaluate performance at scale and establish budgets for next-gen architectures.
• Analyze performance metrics to identify GPU hotspots and root causes, then propose and co-implement solutions with component teams.
• Support teams in porting serial algorithms to GPU to maximize compute utilization and reduce latency.
• Collaborate with the Core team to design a middleware framework promoting efficient CPU and GPU code development.

Requirements

• BS in computer science or related field and 3+ years of experience.
• Strong knowledge of CUDA as applied to recent GPU microarchitectures (e.g., Ampere, Blackwell) and experience debugging/optimizing GPU kernels using tools like Nsight.
• Strong knowledge of C++ and experience in large code bases, comfortable in Linux development environments.
• Experience in development, debugging, and profiling of complex multiprocess systems (e.g., robotic systems, game engines).

Nice to have

• Experience with GPU kernel development in real-time environments, including PTX-level programming, CPU SIMD instructions (e.g., AVX intrinsics), and custom CUDA layers with TensorRT & XLA.
• Hands-on work with ML model optimization (post-training quantization, layer pruning) or hand-tuning GPU kernels (OpenGL, CUDA, RocM).
• Proficiency with SQL, DataBricks, Looker, or other business intelligence tools.

Culture & Benefits

• Comprehensive benefits package including paid time off (sick leave, vacation, bereavement), unpaid time off, health insurance, long-term and short-term disability insurance, life insurance, and long-term care insurance.
• Compensation includes base salary, Amazon RSUs, Zoox Stock Appreciation Rights, and possible sign-on bonus.
• Fast-moving, highly execution-oriented team focused on innovation in autonomous mobility.