A

AMD

Designs chips for computers and AI systems

Software Development Engineer Fellow - GEMM Optimization

Systems Software EngineeringConfirmed live in the last 24 hours
Full-timeSan Jose, CA, USANo salary listedIn-person

About the Job

Requirements

  • Strong command of GPU computer architecture, including compute units, register files, cache and LDS hierarchies, memory bandwidth, matrix cores or WMMA-style instructions, and instruction scheduling and latency hiding.
  • Deep expertise in GEMM algorithms and their mapping onto GPU hardware, including tiling, blocking, register and LDS allocation, wave scheduling, and memory hierarchy utilization.
  • Strong software engineering fundamentals and a track record of building production-quality, high-performance software.
  • Demonstrated experience optimizing GPU compute kernels, ideally GEMM and attention.
  • Proficiency in C++ and assembly-level GPU programming, and working knowledge of Python for tooling and automation.
  • Solid understanding of parallel computing, memory hierarchies, and hardware-software performance tradeoffs.
  • Clear written and verbal communication skills and ability to work effectively with hardware architects, customers, and cross-functional engineering teams.

Responsibilities

  • Analyze AMD GPU hardware specifications in depth and work closely with hardware architects to align kernel design with current silicon capabilities and influence requirements for future generations.
  • Develop support for new instruction-set architecture and hardware/software optimization features in the code generator.
  • Innovate and implement new algorithms for implementing GEMM and GEMM + X.
  • Profile and root-cause GEMM performance bottlenecks across the stack.
  • Partner with customers and internal stakeholders to understand real-world workload requirements, reproduce performance issues, and deliver targeted performance.

Desired Qualifications

  • Direct experience with any GPU compiler backend.
  • Experience with AMD GPU architectures (CDNA/RDNA) or comparable competitive architectures (NVIDIA Hopper/Blackwell, etc.).
  • Contributions to industry standards, publications, or patents related to GPU compute or matrix-multiply optimization.
  • A BS, MS, or PhD in Computer Science or Computer Engineering, or a related field, with deep relevant experience.

About the Company

AMD

Designs chips for computers and AI systems

AMD designs processors, graphics chips and AI accelerators for computer makers, cloud operators and businesses that build electronic devices. Its Ryzen processors power personal computers, while EPYC processors handle work in data centers and Radeon chips render graphics. Instinct accelerators run AI models, paired with software that developers use to train models and put them into production. AMD sells chips and related computing systems through equipment makers, distributors and retail channels, and contracts with outside foundries to fabricate its semiconductor designs. It was founded in 1969 and has its headquarters in Santa Clara, California.

Company size
10,001+
Company stage
IPO
Headquarters
Santa Clara, CA
Founded
1969

Recently Posted Jobs

Explore more opportunities at AMD

See more jobs