Triton Compiler Senior Engineer

AMD - Advanced Micro Devices

Cambridge, United Kingdom

THE PERSON:

An ideal candidate should be familiar with compilers, GPU architectures, parallel programming, and/or high-performance kernels. He/she should be comfortable at performing quantitative analysis of workload and drive improvements across different compiler layers. Most importantly, the candidate is willing to learn and work across boundaries.

KEY RESPONSIBILITIES:

Develop and maintain Triton compiler's AMD backend
Improve various compilation patterns and passes in Triton
Research and author high performance matmul and attention kernels in Triton
Profile kernel performance on AMD GPUs and improve bottlenecks
Fix issues in AMDGPU backend in LLVM

PREFERRED EXPERIENCE:

Familiarity or existing experience with Triton is a strong plus
Familiarity with compiler projects like MLIR, LLVM, IREE
Deep understanding of GPU architectures and programming models
Deep experience with writing high performance GPU kernels and GPU performance tuning
Experience debugging cross-stack issues and reducing user problems to actionable execution
Open-source development ethos