Fellow GPU Kernel Engineer - ML/HPC Optimizations

AMD; Austin (TX); Hybrid; USD 190,000 - 270,000; Full time

Job summary

AMD in Austin, TX is seeking a Fellow to define and drive end-to-end software optimization strategy for top-tier customers. You will shape GPU kernel performance, contribute to ROCm core software, and collaborate across ML workloads and AI frameworks to deliver high-performance solutions.

A successful candidate will have deep experience in compute and machine learning software development, strong collaboration skills, and a track record of delivering optimized GPU kernels in fast-paced

Qualifications

• Bachelor’s, Master’s, or PhD in CS/CE or equivalent required.
• Experience in GPU-centric software optimization and ML workloads.
• Strong collaboration and communication in fast-paced environments.

Responsibilities

• Perform external benchmarking to identify opportunities to improve GPU kernel performance.
• Analyze and optimize GPU kernels across the AI software stack.
• Partner with customers and hyperscalers to tailor architectural optimizations.
• Collaborate with hardware, compiler, and framework teams to influence features.

Skills

GPU kernel optimization

ML software development

Performance analysis

Cross-functional collaboration

GPU architecture

ROCm

Education

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

Tools

ROCm

Job description

AMD in Austin, TX is seeking a Fellow to define and drive end-to-end software optimization strategy for top-tier customers. You will shape GPU kernel performance, contribute to ROCm core software, and collaborate across ML workloads and AI frameworks to deliver high-performance solutions.

A successful candidate will have deep experience in compute and machine learning software development, strong collaboration skills, and a track record of delivering optimized GPU kernels in fast-paced