Senior Triton Kernel & Compiler Engineer for GPUs

Advanced Micro Devices, Inc.; San Jose (CA); Hybrid; USD 180,000 - 260,000; Full time

Benefits offered by this job

AMD benefits

Job summary

Advanced Micro Devices, Inc. is seeking a Senior Triton Compiler and Kernel Engineer to advance Triton performance on AMD GPUs.

You will work across GPU architecture, compilers, kernels, multi-GPU communication, and AI frameworks while contributing to upstream Triton and ROCm. The role requires deep knowledge of GPU architecture, compiler optimization, and high-performance kernel development, with collaboration across teams and mentoring of others.

Qualifications

• Experience with AMD GPU architecture and ROCm is desirable.
• Strong background in compiler development and optimization.
• Proven ability to create and optimize high-performance GPU kernels.

Responsibilities

• Develop and optimize Triton compiler support for AMD GPUs.
• Improve compiler lowering, optimization, scheduling, and code generation.
• Create high-performance kernels for AI workloads like attention and GEMM.
• Develop and optimize multi-GPU kernels and communication primitives.
• Collaborate with architecture, ROCm, PyTorch, and AI framework teams.
• Contribute designs to upstream Triton and LLVM/MLIR.

Skills

GPU architecture

Compiler development

High-performance kernels

Multi-GPU communication

Distributed AI training

AI performance profiling

Education

Bachelor's or Master's degree

Tools

Triton

LLVM

MLIR

CUDA

HIP

NCCL

RCCL

Job description

Advanced Micro Devices, Inc. is seeking a Senior Triton Compiler and Kernel Engineer to advance Triton performance on AMD GPUs.

You will work across GPU architecture, compilers, kernels, multi-GPU communication, and AI frameworks while contributing to upstream Triton and ROCm. The role requires deep knowledge of GPU architecture, compiler optimization, and high-performance kernel development, with collaboration across teams and mentoring of others.