Zoox logo

Senior Software Engineer (C++ GPU Performance)

Zoox

On-siteSeattle, WAsenior$217k–$307kPosted 8h ago

Job description

  • The Software Performance team’s mission is to analyze, optimize and provide guidance to the software and hardware teams in order to meet the required specifications

  • As a GPU performance software engineer within the Software Performance team, you will instrument, monitor, analyze and optimize GPU-based algorithms that are performance-critical for our solution

  • The scope for GPU usage ranges from traditional computer vision and deep learning architectures to complex geometric reasoning and multi-agent decision making. Your work will strongly influence design decisions of future compute platforms & resource allocation

  • Build real-time instrumentation for performance monitoring (CPU, GPU, latency, memory) and develop offline benchmarking frameworks, tools, and scripts to evaluate & analyze performance at scale in CI/vehicle, and establish budgets for next-gen architectures

  • Analyze performance metrics to identify GPU hotspots and root causes, and propose and co-implement actionable solutions with component teams

  • Support teams on bringing serial algorithms to the GPU to maximize compute utilization and improve overall latency

  • Work as part of the Core team to design a middleware framework that promotes by default efficient and performant code development by maximizing CPU and GPU

Benefits

  • Paid parental leave

  • Affinity groups and sports clubs

  • Work from home opportunities

  • Health insurance

  • Our crew’s health and happiness is our first priority. We offer comprehensive health and mental health support, a wellbeing program, and unlimited and flexible paid time away

  • We invest in our crew—and their families—for the long term. That includes generous family planning support, caregiver support, and strong cash compensation with great equity upside

  • We look after our crew when they’re in the office too. Our famous food program is a great example, featuring a daily changing menu of local and sustainable dishes

  • There’s a busy calendar of social events at Zoox, with more sports teams than you can count. And, of course, playing with robots is an important part of the job description- Strong knowledge of C++ and experience in large code bases, comfortable in Linux development environments

  • Strong knowledge of CUDA as applied to recent GPU microarchitectures (e.g., Ampere, Blackwell) and experience debugging/optimizing GPU kernels using tools like Nsight

  • BS in computer science or related field and 7+ years of experience

  • Experience in development, debugging, and profiling of complex multiprocess systems (e.g., robotic systems, game engines)

  • You do not need to match every listed expectation to apply for this position. Here at Zoox, we know that diverse perspectives foster the innovation we need to be successful, and we are committed to building a team that encompasses a variety of backgrounds, experiences, and skills

  • Proficiency with SQL, DataBricks, Looker, or other business intelligence tools

  • Experience with GPU kernel development in a real-time environment, including PTX-level programming, CPU SIMD instructions (e.g., AVX intrinsics), and custom CUDA layers with frameworks like TensorRT & XLA

  • Hands-on work with ML model optimization (post-training quantization, layer pruning, etc) or hand-tuning GPU kernels (in OpenGL, CUDA, RocM or similar)