GPU Kernel Development Engineer - Fellow
Hybrid (Austin, TX)$231.4kFull-timePosted 1mo ago
About the role
Fellow responsible for defining and driving end-to-end software optimization strategy to achieve industry-leading GPU kernel performance, with involvement in GPU architecture feature design, usage, API design, optimization, and deployment for AMD ROCm software stack.
Responsibilities
- Perform external benchmarking and research to define opportunities to improve GPU kernel performance on AMD machine learning products
- Analyze and optimize the performance of GPU kernels, providing insight on optimization paths across the AI software stack
- Partner with top customers and hyperscalers to understand their workload requirements and deliver tailored architectural wins and software optimizations
- Collaborate across hardware architecture, compiler, and framework teams to influence future silicon features based on evolving AI workload trends
- Communicate and collaborate with key technical experts across AMD and with partners and customers to improve ROCm applications, libraries, and tools, as well as hardware
Qualifications
- Bachelor’s or Master’s or PhD degree in Computer Science, Computer Engineering, or equivalent
- Strong understanding of modern model architectures (Transformer, Attention, KV Cache) and optimization techniques like quantization, speculative decoding, and FlashAttention
- Excellent GPU architecture knowledge, performance analysis, algorithm development, and machine learning experience
- Proven history of optimizing GPU kernels with advanced algorithms and across entire AI software stack
- Demonstrated ability to drive cross-functional initiatives in fast-paced, ambiguous environments
Skills
Required:GPU kernel optimizationMachine learning software developmentBenchmarking and performance analysis
Benefits
- AMD benefits at a glance
Last updated Oct 22, 2026. Interested? Apply on the company site.
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
More open roles at AMD
- System Thermal Design Staff Engineer4d agoPenang, Malaysia · Engineering
- PD Engineer4d agoHybrid (Bangalore) · Engineering
- PD Engineer4d agoBangalore · Engineering
- Low-Power & CDC Verification Engineer4d agoSanta Clara, California · Engineering
- Technical Program Manager4d agoShanghai · Engineering