AI Systems & GPU Performance Engineer Intern

Advanced Micro Devices; San Jose, Santa Clara (CA, CA); Hybrid; USD 22,000 - 45,000; Full time

Job summary

Advanced Micro Devices in San Jose, CA is seeking a highly motivated PhD AI Systems & GPU Performance Engineering Intern. You will optimize AI models, training workflows, and inference on AMD Instinct GPUs using ROCm, PyTorch, JAX, and related tools.

The role blends profiling, benchmarking, and end-to-end optimization to advance next-generation AI tech. Join a team that explores cutting-edge AI systems research, collaborates across disciplines, and shapes performance improvements on AMD's latest

Qualifications

• Pursuing a PhD in a technical field with internship end date after the term.
• Strong Python and C/C++ coding experience for accelerator work.
• Familiarity with Linux, profiling, and performance analysis.

Responsibilities

• Profile, benchmark, and optimize AI training and inference workloads on AMD GPUs.
• Develop reproducible benchmarking frameworks and automation scripts.
• Analyze end-to-end AI workflows, including LLMs and mixed precision techniques.
• Collaborate with engineers to evaluate GPU performance across environments.
• Contribute data-driven recommendations to improve efficiency and scalability.

Skills

Python

C/C++

Linux

GPU programming

Profiling

Performance analysis

Education

PhD student in CS/CE/EE

Tools

CUDA

HIP

Triton

rocProfiler

ROCm

Job description

Advanced Micro Devices in San Jose, CA is seeking a highly motivated PhD AI Systems & GPU Performance Engineering Intern. You will optimize AI models, training workflows, and inference on AMD Instinct GPUs using ROCm, PyTorch, JAX, and related tools.

The role blends profiling, benchmarking, and end-to-end optimization to advance next-generation AI tech. Join a team that explores cutting-edge AI systems research, collaborates across disciplines, and shapes performance improvements on AMD's latest