Senior AI GPU & Distributed-Systems Engineer

Advanced Micro Devices, Inc. (AMD); United States; On-site; USD 150,000 - 190,000; Full time

Job summary

Advanced Micro Devices, Inc. (AMD) seeks a Senior Software Engineer - AI to advance the Triton compiler and runtime for the AMDGPU backend.

This full-time role designs distributed execution and communications for large-scale AI workloads across CPU/GPU boundaries, collaborating with compiler, runtime, and hardware teams. The ideal candidate has deep GPU architecture knowledge, experience with Triton/CUDA, and a track record optimizing multi-GPU AI workloads.

Qualifications

• Experience with GPUs, Triton, and multi-GPU systems.
• Experience AI/ML/HPC workloads on large-scale systems.
• Bachelor's or Master's in Computer Engineering, CS or EE.

Responsibilities

• Design and implement native distributed communication and execution for the Triton AMDGPU backend.
• Drive performance optimization across compute and inter-GPU data movement and memory hierarchy.
• Collaborate with GPU architecture and performance teams to co-design distributed GPU programming capabilities.

Skills

GPU architecture

Compiler tech

Distributed GPU systems

Triton/CUDA

Performance optimization

Education

CS/EE/CE degree

Tools

HIP

Job description

Advanced Micro Devices, Inc. (AMD) seeks a Senior Software Engineer - AI to advance the Triton compiler and runtime for the AMDGPU backend.

This full-time role designs distributed execution and communications for large-scale AI workloads across CPU/GPU boundaries, collaborating with compiler, runtime, and hardware teams. The ideal candidate has deep GPU architecture knowledge, experience with Triton/CUDA, and a track record optimizing multi-GPU AI workloads.