Fellow, GPU Kernel Optimization for AI & ML

Advanced Micro Devices, Inc.; Austin (TX); On-site; USD 190,000 - 300,000; Full time

Benefits offered by this job

AMD benefits at a glance

Job summary

Advanced Micro Devices, Inc. is seeking a Fellow to define and drive end-to-end software optimization for top-tier customers. You will influence GPU kernel design, ROCm features, and ML workloads while collaborating with AMD partners and customers.

The role emphasizes performance, architectural insight, and cross-team coordination to deliver high-impact ROcm-based solutions for AI and HPC workloads.

Qualifications

• Experience in compute and ML software to deliver high-performance ROCm core software.
• Strong collaboration and willingness to learn and innovate in ML and HPC on GPUs.
• Ability to drive optimization across the AI software stack.

Responsibilities

• Perform external benchmarking and research to identify GPU kernel improvements for ML products.
• Analyze and optimize GPU kernel performance; provide insights across the AI stack.

Skills

GPU kernel development

ML software development

ROCm

Performance optimization

Cross-functional collaboration

Architectural design

Transformer architectures

Education

Bachelor’s or Master’s or PhD degree in CS/CE

Tools

ROCm

GPU architecture

Job description

Advanced Micro Devices, Inc. is seeking a Fellow to define and drive end-to-end software optimization for top-tier customers. You will influence GPU kernel design, ROCm features, and ML workloads while collaborating with AMD partners and customers.

The role emphasizes performance, architectural insight, and cross-team coordination to deliver high-impact ROcm-based solutions for AI and HPC workloads.