← All roles

Triton Compiler and Kernel Software Engineer

AMD· Engineering
Apply on company site
San Jose, CA$204k - $306kFull-timePosted 3w ago

About the role

Senior Triton Compiler and Kernel Engineer to advance Triton performance and capabilities on AMD GPUs, contributing across GPU architecture, compilers, kernels, multi-GPU communication, and AI frameworks within AMD’s ROCm ecosystem.

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.

Qualifications

  • Bachelor’s or Master’s degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent

Skills

Required:Triton compiler developmentGPU architecture and programmingKernel optimization
Nice to have:Distributed training/inferenceMulti-GPU communication

Benefits

  • Benefits offered are described: AMD benefits at a glance.
Last updated Oct 22, 2026. Interested? Apply on the company site.
AMD logo

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
Company profileWebsite

More open roles at AMD