← All roles

Fellow GPU Performance Optimization Engineer

AMD· Engineering
Apply on company site
Hybrid (San Jose, CA)$268kFull-timePosted 6mo ago

About the role

Fellow GPU Performance Optimization Engineer on the Models and Applications team to maximize performance and efficiency of large-scale AI training workloads on AMD GPU platforms, driving cross-stack optimizations and influencing architecture and software stack design.

Responsibilities

  • Lead performance optimization of large-scale AI training workloads on AMD GPU platforms across single-node and multi-node environments.
  • Identify and eliminate system bottlenecks across compute, memory, and communication (e.g., kernel efficiency, memory bandwidth, network utilization).
  • Optimize distributed training strategies (Data, Tensor, Pipeline Parallelism, ZeRO, etc.) for scalability and efficiency on AMD hardware.
  • Drive cross-stack optimizations spanning kernels, compilers, runtimes, communication libraries, and ML frameworks.
  • Develop and apply advanced profiling, benchmarking, and performance modeling methodologies.
  • Collaborate with hardware, compiler, and framework teams to influence next-generation GPU architecture and software stack design.
  • Contribute to and lead open-source efforts to improve ecosystem performance on AMD platforms.
  • Define best practices and guide teams on performance tuning for large-scale training workloads.
  • Stay at the forefront of advancements in large-scale training systems and performance optimization techniques.

Qualifications

  • Ph.D. in Computer Science, Computer Engineering, or a related field preferred, or equivalent industry experience with significant technical impact.

Skills

Required:GPU performance optimizationGPU architectureProfiling and bottleneck analysisDistributed training experienceDistributed training frameworksCommunication librariesML frameworksProgramming languagesTechnical leadership
Nice to have:Compiler/kernels optimization

Benefits

  • Benefits offered are described: AMD benefits at a glance.
Last updated Oct 15, 2026. Interested? Apply on the company site.
AMD logo

About AMD

together we advance_AI

AMD delivers leadership high-performance and adaptive computing solutions to advance data center AI, AI PCs, intelligent edge devices, gaming, & beyond.

AIData CenterHigh-Performance ComputingGamingEdge Devices
Company profileWebsite

More open roles at AMD