Sr. Software Engineer - AI Triton Communication

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

Benefits offered by this job

AMD benefits

Job summary

AMD is seeking an experienced engineer to advance the Triton compiler and runtime for AMD Instinct GPUs. You will design native distributed execution, optimize inter-GPU communication, and work across the compiler, runtime, and ROCm layers to push scalability and performance.

Collaboration with architecture and performance teams is essential as AMD broadens its multi-GPU AI capabilities. The role focuses on delivering industry-leading distributed GPU support, contributing to open-source

Qualifications

• Experience in GPU architecture, compiler tech, and distributed GPU systems at multi-GPU scale.
• Ability to work across compiler, runtime, and hardware layers to maximize scalability.
• Background in optimizing AI/HPC workloads on multi-GPU platforms.

Responsibilities

• Design and implement native distributed communication and execution capabilities within the Triton AMDGPU backend for scalable multi-GPU execution.
• Develop compiler and runtime mechanisms for native GPU-initiated communication including collectives and remote memory access.
• Profile, debug, and optimize cross-stack performance across compiler, runtime, ROCm, and hardware.
• Collaborate with architecture, compiler, runtime, and performance teams to enable next-gen distributed GPU programming.
• Contribute to open-source Triton and ROCm distributed ecosystem.

Skills

GPU architecture

Compiler technologies

Distributed GPU systems

Multi-GPU performance

Performance tooling

Education

Bachelor’s or Master’s Degree in Computer Engineering, Computer Science, Electrical Engineering

Tools

Triton

ROCm

NCCL

NVSHMEM

MPI

Job description

ADVANCE YOUR CAREER. ADVANCE THE WORLD.

At AMD, we believe technology can change lives for the better. It can heal us, entertain us, and make us more connected, productive, and understanding of the world around us. And we’re looking for talent who feel the same: people who want to leave the planet better than they found it, those who don’t shy away from humanity’s challenges but are determined to help solve them.

AMD is powering the next generation of supercomputing, high-performance computing, cloud, and AI. Whether you’re designing next-gen processors, enabling AI breakthroughs, or creating go-to-market plans, every role at AMD contributes to something bigger — technology that moves the world forward.

THE ROLE:

Triton is a widely adopted language and compiler for high-performance GPU kernels, powering major AI frameworks such as PyTorch, vLLM, and SGLang. As AI workloads increasingly scale across multiple GPUs and nodes, first-class support for distributed execution and communication in Triton is strategically critical to enabling efficient large-scale training and inference on AMD Instinct Accelerators. AMD GPUs are an official Triton backend, and delivering industry-leading distributed performance and scalability on AMD Instinct accelerators is a key priority. The performance, scalability, and usability of Triton directly impact the competitiveness of AMD hardware in large-scale AI deployments. In this role, you will advance the Triton compiler and runtime stack for AMD CDNA and next-generation GPU architectures by building native distributed execution and communication capabilities. You will develop compiler and runtime infrastructure that enables efficient inter-GPU communication, scalable execution, and optimal hardware utilization. You will work across compiler, runtime, and hardware layers, collaborating closely with GPU architecture and software teams to help establish AMD GPUs as a best-in-class platform for Triton-based distributed AI.

THE PERSON:

The ideal candidate has deep expertise in GPU architecture, compiler technologies, and distributed GPU systems, with proven experience optimizing workloads at multi-GPU scale. You are comfortable working across the full execution stack — from compiler and runtime to hardware — and understand how GPU execution, memory hierarchy, and inter-GPU communication impact performance. You have experience working close to the GPU runtime, communication stack, or compiler backend, and are motivated to build native distributed execution and communication capabilities tightly integrated with the compiler and runtime to maximize scalability and hardware utilization. You thrive on solving complex system-level performance challenges and delivering scalable, high-performance GPU infrastructure.

KEY RESPONSIBILITIES:

• Design and develop native distributed communication and execution capabilities within the Triton AMDGPU backend, enabling scalable multi-GPU execution for large-scale AI workloads
• Design and implement Triton compiler and runtime mechanisms for native GPU-initiated communication, including collective operations, remote memory access, synchronization, and distributed execution primitives
• Drive performance optimization across compute and communication, including inter-GPU data movement, communication/computation overlap, memory hierarchy utilization, and GPU-driven scheduling efficiency
• Develop and optimize distributed Triton kernels and execution models to achieve high performance, scalability, and efficient hardware utilization for AI workloads
• Analyze, profile and debug complex cross-stack issues spanning Triton compiler, runtime, ROCm stack, and GPU hardware execution

• Collaborate closely with GPU architecture, compiler, runtime, and performance teams to co-design and enable next-generation distributed GPU programming and execution capabilities
• Contribute to open-source Triton and ROCm distributed ecosystem, driving innovation in distributed GPU computing

PREFERRED EXPERIENCE:

• Deep experience in compiler development, GPU software, distributed systems, or performance engineering

• Familiarity or hands-on experience with Triton compiler and runtime
• Deep understanding of modern GPU architectures, including execution model, memory hierarchy (LDS, L2, HBM), scheduling, occupancy, and hardware performance characteristics
• Good understanding of GPU runtime systems, communication stacks, and multi-GPU interconnects such as XGMI, NVLink, PCIe, or InfiniBand and their performance implications
• Familiarity with distributed GPU communication libraries such as RCCL, NCCL, NVSHMEM, rocSHMEM, or MPI and similar technologies
• Experience developing, optimizing, and scaling workloads across multiple GPUs, including inter-GPU communication, synchronization, and communication/computation overlap

• Strong experience with GPU programming using Triton, HIP, CUDA, or similar parallel programming environments
• Strong knowledge of MLIR and/or LLVM internals
• Experience profiling, debugging, and optimizing performance across compiler, runtime, and hardware layers
• Familiarity with ROCm, HIP, CUDA, or similar GPU programming ecosystems, including performance profiling and optimization tools
• Experience optimizing large-scale AI, machine learning or HPC workloads across multi-GPU systems
• Experience contributing to open-source projects and working in collaborative, cross-functional engineering environments
• Strong problem-solving, communication, and technical leadership skills

PREFERRED ACADEMIC CREDENTIALS:

• Bachelor’s or Master's Degree in Computer Engineering, Computer Science, Electrical Engineering or equivalent practical experience

This role is not eligible for visa sponsorship.

• Benefits offered are described: AMD benefits at a glance.

AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third-party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law. We encourage applications from all qualified candidates and will accommodate applicants’ needs under the respective laws throughout all stages of the recruitment and selection process.

AMD may use Artificial Intelligence to help screen, assess or select applicants for this position. AMD’s “Responsible AI Policy” is available here.

This posting is for an existing vacancy.