Expert AI Kernel Developer

About

Join AMD as an Expert AI Kernel Developer and transform AI workloads with high-performance Triton kernels. This role focuses on leveraging advanced GPU architecture for groundbreaking computing. As a Senior Software Engineer, you will play a significant role in creating optimized Triton and Gluon kernels crucial for AI model performance. Your collaboration with researchers, compiler engineers, and hardware teams will enhance AMD's leadership in AI technologies while directly impacting product efficiency and scalability.

Responsibilities

• Implement high-performance matmul and attention kernels with Triton
• Ensure correctness and scalability for Triton kernels
• Maintain the Triton AMD backend in ROCm and LLVM
• Drive optimizations across memory hierarchy and execution
• Conduct profiling to identify and resolve performance issues

Requirements

• Minimum of 3 years in GPU kernel development
• Solid experience with Triton and AI frameworks
• Deep knowledge of GPU execution and memory management
• Proven contributions to open-source AI/ML projects
• Bachelor’s or Master’s in a technical field

Bring your technical expertise and creativity to AMD and advance AI technology.