Senior GPU Software Engineer - HPC & ML Libraries

Advanced Micro Devices, Inc.; San Jose (CA); On-site; USD 210,000 - 280,000; Full time

Job summary

Advanced Micro Devices, Inc. in San Jose, CA is seeking a Senior Member of Technical Staff to advance GPU libraries within the ROCm ecosystem.

You will collaborate across library teams to design, implement, validate, and deliver core software capabilities that integrate with GPU kernels in broader systems. The role emphasizes architecture-driven software development, mentorship of junior staff, and working across hardware, firmware, and software teams to translate complex ideas into robust,

Qualifications

• Hands-on experience with C++ and GPUs, including CUDA/HIP/OpenMP/MPI/OpenCL for accelerated computing.
• Proficiency in Python and AI/ML frameworks is a plus.
• Experience contributing to high-performance software systems with CI/CD and orchestration tooling.

Responsibilities

• Design, implement, validate, and deliver software library capabilities for GPU kernels within ROCm.
• Influence architecture and guide technical execution across library teams.
• Collaborate with staff to develop robust designs considering performance and roadmap commitments.
• Identify gaps and remediate with library owners and engineering colleagues.
• Mentor senior and principal staff and promote best practices in the agile lifecycle.

Skills

C++

CUDA

HIP

Python

Education

Master's degree

Tools

Jenkins

Docker

Kubernetes

Job description

Advanced Micro Devices, Inc. in San Jose, CA is seeking a Senior Member of Technical Staff to advance GPU libraries within the ROCm ecosystem.

You will collaborate across library teams to design, implement, validate, and deliver core software capabilities that integrate with GPU kernels in broader systems. The role emphasizes architecture-driven software development, mentorship of junior staff, and working across hardware, firmware, and software teams to translate complex ideas into robust,