Technical Program Manager – ROCm Libraries

THE ROLE:

Do you want to drive end-to-end delivery of artificial intelligence, math, computer vision, and communication libraries to enable high performance computing and artificial intelligence? AMD is searching for a talented and motivated Program Manager to join the GPU libraries team developing Math Libraries as part of the AMD ROCm™ Open Software Platform.

THE PERSON:

You are accustomed to working in a dynamic, geographically distributed agile team, where partnership and collaboration are paramount. You possess excellent written and verbal communication skills, strong organization, and attention to detail. With a keen interest in data, you will draw on your strong technical background, analytical capabilities, and interpersonal skills to drive process improvements, improve program operations, and engage with key partners at all levels, from driving technical discussions with developers through to communication with executives.

KEY RESPONSIBILITIES:

• As a TPM in this group, you will not just track schedules; you will be an active technical partner. You will facilitate architectural decision-making, assist with the triage and debugging of complex issues, and translate intricate technical requirements for diverse stakeholders ranging from kernel engineers to executive leadership. You will operate at the intersection of High-Performance Computing (HPC) and Artificial Intelligence (AI), ensuring our math libraries deliver world-class performance on AMD Instinct™ and AMD Radeon™ accelerators.

• 
  Program Execution & Delivery: Drive the end-to-end lifecycle of the ROCm Libraries including release planning, roadmap definition, and execution for Libraries such as rocBLAS, hipBLASLt, and other Libraries as required.

• 
  Compiler & Code Gen: Facilitate discussions on code generation strategies (e.g., auto-tuning kernels) and optimizations at the Intermediate Representation (IR) level to maximize hardware utilization.

• 
  Technical Triage & Debugging: Actively participate in the triage process for software defects. Leverage your technical background to help engineers prioritize bugs, understand root causes (whether in the library or the compiler backend), and unblock critical paths.

• 
  Architectural Decision Support: Facilitate technical discussions between software architects, research teams, and silicon engineers. Help drive consensus on API designs and optimization strategies for new hardware generations.

• 
  Stakeholder Management: Translate complex technical requirements and status updates into clear, actionable communications. Bridge the gap between technical engineering teams and management.

Components you will support will include, but are not limited to:

• rocBLAS: The foundational BLAS library implemented in HIP C++ and optimized for AMD GPUs, covering Level-1, Level-2, and Level-3 linear algebra operations.

• hipBLASLt: A high-performance library providing flexible General Matrix-Matrix (GEMM) operations, essential for modern AI/ML workloads that require functionality beyond traditional BLAS (e.g., fusion, post-processing).

• hipSPARSELt: A sparse marshalling library that enables high-performance sparse matrix operations on AMD discrete GPUs, critical for optimizing memory bandwidth in large-scale AI models.

PREFERRED EXPERIENCE:

• Program Management: 5+ years of experience in Technical Program Management, Engineering Management, or as a Senior Software Engineer with leadership responsibilities.

• 
  Compiler & Architecture Knowledge: Experience working with or managing projects involving compiler technologies (e.g., LLVM, GCC, Open64).

• 
  Understanding of Code Generation techniques, JIT compilation, or Intermediate Representations (IR) (e.g., MLIR, LLVM IR) and how they impact library performance.

• 
  GPU & HPC Domain: Deep understanding of GPU computing (AMD ROCm, CUDA, or OpenCL).

• Familiarity with High-Performance Computing (HPC) and Artificial Intelligence (AI) workloads.

• Knowledge of linear algebra concepts (GEMM, Sparse Matrices, Tensor operations).

• Technical Literacy: Ability to read and understand technical documentation, bug reports, and basic code structures (C++, Python, Assembly/ISA familiarity is a plus).

• Process & Tools: Proficiency with Agile/Scrum methodologies and tools like Jira, Confluence, and GitHub.

• Library Development: Hands-on experience developing or managing math libraries (BLAS, FFT, LAPACK) or similar performance-critical software.

ACADEMIC CREDENTIALS:

• Bachelor’s or Master’s degree in Computer Science, Software Engineering, Electrical Engineering, Mathematics, or equivalent strongly preferred

• Certifications such as the PMP or agile certification would be an asset

• Design and develop new AMD software technologies.
• Participate in bringing new GPU products to market.
• Enhancing our existing software to take advantage of new hardware capabilities.
  
• Debug and fix defects in our existing products.
• Work with AMD’s architecture specialists to improve future products.

• 5+ years of professional software development experience.
• Experience with modern software development practices and tools.
• Proficiency in C/C++ and Python.
• Experience with GPU programming using HIP, CUDA, or other similar programming models; or with other concurrent or parallel architectures.df
• Experience with low-level programming and code micro-optimization.

• Bachelor’s or Master's degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent