GPU Software Engineer

Waymo’s Compute Team is tasked with a critical and exciting mission: We deliver the compute platform responsible for running the fully autonomous vehicle’s software stack To achieve our mission, we architect and create high-performance custom silicon; we develop system-level compute architectures that push the boundaries of performance, power, and latency; and we collaborate closely with many other teammates to ensure we design and optimize hardware and software for maximum performance We are a multidisciplinary team seeking curious and talented teammates to work on one of the world’s highest performance automotive compute platforms In this hybrid role, you will report to a Senior Software Engineer Develop high-performance GPU primitives and abstractions to enable Waymo to scale its accelerator codebase across diverse GPU backends Collaborate with Waymo’s internal hardware team and external partners on SoC projects with a focus on the GPU portion Manage the bring-up, correctness, and performance of the Waymo onboard stack on new GPU platforms Contribute to testing infrastructure that enhances the CI/CD flow for GPUs, detects bugs early, and generates automated alerts to maintain the GPU stack’s functionality and performance Create profiler and debugger tools for new GPU platforms Benefits Medical, dental, and vision insurance for employees and dependents Employee assistance programs focused on mental health Personalized workplace adjustments for diverse needs and abilities, including physical, mental, and neurodivergent considerations Access to mental health apps Onsite wellness centers Second medical opinion for you and your loved ones Medical advocacy program for transgender employees Support programs including menopause benefit Counseling services Competitive compensation Regular bonus and equity performance grant opportunities Generous 401(k) and regional retirement plans 1-on-1 financial coaching Annual cross-company compensation review and pay equity analysis Fertility and growing family assistance Parental leave and baby bonding leave Survivor income benefit Backup childcare Elder care and support Caregiver leave Paid time off, including vacation, bereavement, sick leave, parental leave, disability, and holidays Jury duty leave Military leave Hybrid work model with remote work opportunities also available Educational reimbursement Peer learning and coaching platform Donation matching programs Volunteer hours Employee resource groups Internal community groups and local culture clubs Inspiring spaces to work, recharge, and collaborate with fellow Waymonauts On-site meals and snacks Fitness centers, massage programs, and ergonomic support On-demand fitness, wellbeing, and cooking classes Commuter benefits Qualifications Proven expertise in C++ programming Bachelor degrees in EECS, coupled with a minimum of three years of industry experience Proficiency in utilizing performance analysis tools and debuggers Solid understanding of GPU hardware architecture Enthusiasm for developing the complete GPU software stack, from the hardware level to real-world applications Experience with full-system simulation frameworks (SystemC, Gem5 or similar) Knowledge of Linux device drivers and embedded firmware Experience with diverse GPU deployment environments Familiarity with GPU libraries such as Thrust, CUB, CUTLASS, or Eigen Proficiency in GPU optimization techniques, including memory coalescing, register/shared memory tiling, pinned memory, and warp-level programming Experience collaborating with external operators to ensure high quality standards Experience contributing to open-source compiler projects such as LLVM or SPIR-V #J-18808-Ljbffr