Senior AI Triton & Distributed GPU Engineer

AMD; San Jose (CA); On-site; USD 180,000 - 240,000; Full time

Benefits offered by this job

Benefits at a glance

Job summary

AMD is seeking a highly skilled compiler and GPU runtime specialist to advance the Triton backend and distributed execution for AMD Instinct accelerators. You will design native multi-GPU communication, optimize performance, and collaborate with architecture and software teams.

You will work across compiler, runtime, and hardware layers to ensure scalable, high-performance AI workloads on AMD GPUs and contribute to open-source Triton and ROCm ecosystems.

Qualifications

• Deep expertise in compiler development, GPU software, distributed systems, or performance engineering.
• Experience with Triton compiler and runtime is highly desirable.
• Strong understanding of modern GPU architectures, memory hierarchy, and inter-GPU communication.

Responsibilities

• Design and develop native distributed communication and execution within the Triton AMDGPU backend for scalable multi-GPU AI workloads.
• Implement Triton compiler and runtime mechanisms for native GPU-initiated communication and distributed execution primitives.
• Drive performance optimization across compute and communication, including inter-GPU data movement and memory hierarchy utilization.
• Develop and optimize distributed Triton kernels and execution models for AI workloads on AMD GPUs.
• Profile, debug, and resolve cross-stack issues spanning Triton, ROCm, and GPU hardware execution.
• Collaborate with GPU architecture, compiler, runtime, and performance teams to enable next-gen distributed GPU programming.

Skills

Compiler development

GPU software

Distributed systems

Performance engineering

Triton workloads

Open-source contributions

GPU programming

Education

Bachelor’s or Master’s in CS/CE/EE

Tools

Triton

HIP/CUDA

LLVM/MLIR

ROCm

NVLink/XGMI

Job description

AMD is seeking a highly skilled compiler and GPU runtime specialist to advance the Triton backend and distributed execution for AMD Instinct accelerators. You will design native multi-GPU communication, optimize performance, and collaborate with architecture and software teams.

You will work across compiler, runtime, and hardware layers to ensure scalable, high-performance AI workloads on AMD GPUs and contribute to open-source Triton and ROCm ecosystems.