Senior Software Engineer (Planner GPU Compute)

Join Zoox, a company dedicated to revolutionizing transportation. As a Senior Software Engineer on the Planner Compute team, you will optimize GPU-based algorithms critical to our solution. Your expertise in GPU performance and CUDA will support the adaptation of our planner to multiple GPU architectures. Enjoy a range of benefits, including paid parental leave, health insurance, and a vibrant work culture.

• Instrument, monitor, analyze, and optimize GPU-based algorithms that are performance-critical for the solution.
• Analyze performance metrics to identify GPU hotspots and optimizations, and contribute to the adaptation of the current planner to multiple GPU architectures.
• Support engineers within Planner as a subject matter expert on CUDA & GPU performance, and optimize ML models for latency and GPU memory usage.

• BS in computer science or related field and 8+ years of experience
• Experience in development, debugging, and profiling of complex multiprocess systems (e.g., robotic systems, game engines)
• 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
• GPU kernel development experience