Sr. Software Engineer - AI Triton Communication
Advanced Micro Devices, Inc · Posted Nov 11, 2026
- San Jose, California, United States
- On-site
- Full-time · Senior
Advance the Triton compiler and runtime stack for AMD CDNA and next-generation GPU architectures by building native distributed execution and communication capabilities.
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
Requirements
- Bachelor’s Degree in Computer Engineering
- Master's Degree in Computer Engineering
- Bachelor’s Degree in Computer Science
- Master's Degree in Computer Science
- Bachelor’s Degree in Electrical Engineering