Senior AI/ML Compiler Engineer - GPU

Advanced Micro Devices; San Jose (CA); On-site; USD 190,000 - 230,000; Full time

Benefits offered by this job

Benefits at a glance

Equal opportunity employer

Job summary

Advanced Micro Devices is seeking a PMTS AI/ML Compiler Engineer to drive compiler optimizations for AI training and inference on AMD GPU platforms. You will work at the intersection of compiler tech, ML frameworks, and GPU architecture, contributing to ROCm and ONNX Runtime execution stacks.

You will collaborate with architecture, kernel, runtime, and framework teams to deliver high-performance AI software, optimize MLIR/LLVM pipelines, and improve performance across AMD GPUs.

Qualifications

• Strong proficiency in C++ and object‑oriented software development.
• Hands‑on experience with GPU programming technologies such as HIP, CUDA, Triton, or equivalent.
• Experience with compiler technologies including MLIR, LLVM, or related compiler infrastructures.
• Strong understanding of machine learning frameworks, model execution pipelines, and AI workload optimization.

Responsibilities

• Design, develop, and optimize compiler transformations and operator lowerings within MLIR, LLVM, and ROCm software stack.
• Develop scalable solutions that improve portability and performance across AMD GPU architectures.
• Collaborate with compiler and runtime teams to develop advanced optimization strategies for AI workloads.
• Analyze workload execution characteristics and identify opportunities for improvements in compute efficiency and memory usage.
• Lead performance optimization efforts for AI training, inference, and RL workloads on AMD GPUs.

Skills

C++

GPU programming

MLIR/LLVM

AI workload optimization

Linux development

Performance profiling

Team collaboration

Education

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

Tools

HIP

CUDA

Triton

LLVM tooling

ONNX

Job description

Advanced Micro Devices is seeking a PMTS AI/ML Compiler Engineer to drive compiler optimizations for AI training and inference on AMD GPU platforms. You will work at the intersection of compiler tech, ML frameworks, and GPU architecture, contributing to ROCm and ONNX Runtime execution stacks.

You will collaborate with architecture, kernel, runtime, and framework teams to deliver high-performance AI software, optimize MLIR/LLVM pipelines, and improve performance across AMD GPUs.