1 day ago
San Jose, CA, USASenior
Responsibilities
- Develop and optimize Triton compiler support for AMD GPUs, including 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 Triton abstractions.
- Analyze compute, memory, communication, occupancy, register usage, and generated code.
- Resolve 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.
Requirements
- Bachelor’s or master’s degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent.
- Strong experience in several areas including GPU architecture and programming, compiler development and optimization, high-performance GPU kernels, multi-GPU communication, distributed AI training and inference, AI workload performance, and low-level performance analysis.
- Experience with AMD GPU architecture and ROCm is highly desirable.
- Experience optimizing kernels with Triton, HIP, CUDA, or GPU assembly.
- Experience with Triton, LLVM, MLIR, or another optimizing compiler.
- Knowledge of GPU execution models, memory hierarchies, synchronization, and instruction pipelines.
- Experience with collective communication, distributed programming, and libraries such as RCCL or NCCL.
- Familiarity with distributed training, inference, tensor parallelism, expert parallelism, or pipeline parallelism.
- Experience using AI coding tools and autonomous agents for software development, debugging, benchmarking, and kernel performance tuning.
- Familiarity with AI primitives, reduced-precision formats, and performance profiling.
- Contributions to complex or open-source software projects, along with strong analytical, debugging, communication, and collaboration skills.
Benefits
- Benefits are described in AMD’s benefits-at-a-glance materials.
- The role is hybrid, as indicated by the posting.
About AMD
AMD designs and sells CPUs, GPUs, and adaptive/embedded computing products for PCs, data centers, gaming, and edge devices. Its portfolio includes Ryzen and EPYC processors, Radeon and Instinct graphics, and adaptive SoCs from its Xilinx acquisition, sold to OEMs, cloud providers, and device makers. Founded in 1969 and headquartered in Santa Clara, it is a public company on NASDAQ and supplies semi-custom chips for major game consoles.