GPU Software Engineer (Autonomous Driving)

TL;DR

GPU Software Engineer (Autonomous Driving): Develop high-performance GPU primitives and abstractions to scale accelerator codebase across diverse GPU backends with an accent on system-level compute architectures for fully autonomous vehicles. Focus on collaborating with hardware teams on SoC projects, managing bring-up and performance of onboard stack, and creating profiler/debugger tools for new GPU platforms.

Location: Hybrid/Onsite in Mountain View, California or New York City, New York

Salary Range: $204,000—$259,000 USD

Company

Autonomous driving technology company building the Waymo Driver for ride-hail service and various vehicle platforms, with over 100 million autonomous miles driven.

What you will do

• Develop high-performance GPU primitives and abstractions to scale accelerator codebase across diverse GPU backends
• Collaborate with internal hardware team and external partners on SoC projects focusing on GPU
• Manage bring-up, correctness, and performance of Waymo onboard stack on new GPU platforms
• Contribute to testing infrastructure for CI/CD flow, early bug detection, and automated alerts
• Create profiler and debugger tools for new GPU platforms

Requirements

• Bachelor's degree in EECS with minimum 3 years industry experience
• Proven expertise in C++ programming
• Experience with full-system simulation frameworks (SystemC, Gem5 or similar)
• Solid understanding of GPU hardware architecture
• Proficiency in performance analysis tools and debuggers
• Enthusiasm for developing complete GPU software stack from hardware to applications

Nice to have

• Knowledge of Linux device drivers and embedded firmware
• Experience with diverse GPU deployment environments
• Proficiency in GPU optimization techniques (memory coalescing, register/shared memory tiling, pinned memory, warp-level programming)
• Familiarity with GPU libraries (Thrust, CUB, CUTLASS, Eigen)
• Experience collaborating with external operators for high quality standards
• Experience contributing to open-source compiler projects (LLVM, SPIR-V)

Culture & Benefits

• Discretionary annual bonus program
• Equity incentive plan
• Generous company benefits program