Zoox is looking to solve the problem of optimizing the performance of its self-driving hardware and software solution, requiring expert fine-tuning of compute hardware architecture and algorithms to achieve maximum throughput at optimal power levels.
Requirements
- 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)
- 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).
- Proficiency with SQL, DataBricks, Looker, or other business intelligence tools.
Responsibilities
- 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.
Other
- BS in computer science or related field and 7+ years of experience.
- Paid time off (e.g. sick leave, vacation, bereavement), unpaid time off, Zoox Stock Appreciation Rights, Amazon RSUs, health insurance, long-term care insurance, long-term and short-term disability insurance, and life insurance.
- If you need an accommodation to participate in the application or interview process please reach out to [email protected] or your assigned recruiter.
- You do not need to match every listed expectation to apply for this position.
- Zoox is committed to building a team that encompasses a variety of backgrounds, experiences, and skills.
