PhD AI Systems & GPU Performance Intern

Advanced Micro Devices, Inc.; Santa Clara (CA); Hybrid; USD 75,000 - 105,000; Full time

Job summary

Advanced Micro Devices, Inc. seeks a PhD AI Systems & GPU Performance Engineering Intern to optimize AI workloads on AMD Instinct GPUs using ROCm, PyTorch, JAX, and related tools.

This role places you at the heart of AMD's AI ecosystem, contributing to profiling, benchmarking, and end-to-end performance improvements. You will develop reproducible benchmarks, automate performance validation, and work with engineers to deliver data-driven recommendations that enhance efficiency, scalability, and

Qualifications

• Currently pursuing a PhD in Computer Science, Computer Engineering, Electrical Engineering, Applied Mathematics, or related field with graduation after internship.
• Hands-on programming in Python and C/C++, with exposure to CUDA, HIP, Triton, or similar frameworks.
• Experience with PyTorch, JAX, or TensorFlow and large-scale AI model concepts.

Responsibilities

• Profile, benchmark, and optimize AI training and inference workloads on AMD Instinct GPUs.
• Develop reproducible benchmarking frameworks and automation scripts for performance validation.
• Collaborate with engineers to evaluate GPU performance and provide data-driven recommendations.

Skills

Python

C/C++

CUDA

HIP

Triton

Education

PhD candidate (CS/CE/EE)

Tools

ROC Profiler

ROCm

Nsight

PyTorch

JAX

Job description

Advanced Micro Devices, Inc. seeks a PhD AI Systems & GPU Performance Engineering Intern to optimize AI workloads on AMD Instinct GPUs using ROCm, PyTorch, JAX, and related tools.

This role places you at the heart of AMD's AI ecosystem, contributing to profiling, benchmarking, and end-to-end performance improvements. You will develop reproducible benchmarks, automate performance validation, and work with engineers to deliver data-driven recommendations that enhance efficiency, scalability, and