Triton Compiler and Kernel Software Engineer

AMD is seeking a Senior Triton Compiler and Kernel Engineer to advance Triton performance and capabilities on AMD GPUs. The role focuses on compiler development and optimization, high-performance AI kernels, multi-GPU communication, and contributions to upstream Triton and AMD’s ROCm software stack.

Responsibilities:

• Develop and optimize Triton compiler support for AMD GPUs
• Improve compiler lowering, optimization, scheduling, and code generation
• Create high-performance kernels for attention, GEMM, MoE, and other AI workloads
• Develop and optimize multi-GPU kernels and communication primitives
• Enable new AMD GPU and interconnect capabilities through effective Triton abstractions
• Analyze compute, memory, communication, occupancy, register usage, and generated code
• Resolve complex correctness and performance issues across single- and multi-GPU workloads
• Collaborate with GPU architecture, ROCm, PyTorch, and AI framework teams
• Contribute designs and implementations to upstream Triton and LLVM/MLIR
• Provide technical leadership, code reviews, and mentoring