Senior AI Frameworks & GPU Kernel Engineer

AMD; Santa Clara (CA); On-site; USD 180,000 - 240,000; Full time

Benefits offered by this job

Benefits at a glance

Job summary

AMD is seeking a senior C++/Linux expert to optimize and develop deep learning frameworks for AMD GPUs. You will work on GPU kernels, model performance, and multi-GPU/multi-node scales while collaborating with internal teams and open‑source maintainers.

The ideal candidate combines deep learning experience with advanced compiler and HPC knowledge, and will provide technical leadership across projects and teams.

Qualifications

• Experience designing and optimizing GPU kernels for deep learning on AMD GPUs.
• Proven ability to integrate GPU-accelerated compute into ML frameworks (PyTorch, TensorFlow).
• Advanced proficiency in Python and C++ with performance tuning and robust test design.

Responsibilities

• Optimize Deep Learning Frameworks: enhance frameworks for AMD GPUs in open‑source repos.
• Develop GPU kernels to maximize AI operation performance.
• Collaborate with internal GPU library teams and open‑source maintainers for upstream integration.

Skills

C++ development

Linux

Deep learning

GPU optimization

Team leadership

Python

Education

Master’s/PhD in CS/CE/EE

Tools

HIP

CUDA

CK

CUTLASS

Triton

LLVM

ROCm

Job description

AMD is seeking a senior C++/Linux expert to optimize and develop deep learning frameworks for AMD GPUs. You will work on GPU kernels, model performance, and multi-GPU/multi-node scales while collaborating with internal teams and open‑source maintainers.

The ideal candidate combines deep learning experience with advanced compiler and HPC knowledge, and will provide technical leadership across projects and teams.