2027 PhD AI Training Systems and Performance Engineer Intern/Co-Op

Advanced Micro Devices, Inc.; San Jose (CA); Hybrid; USD 83,000 - 104,000; Full time

Job summary

Advanced Micro Devices, Inc. seeks a highly motivated AI Training Systems & Performance Engineering PhD Intern/Co-Op to join our team in the San Jose area.

You will help accelerate AI training workloads on AMD Instinct GPUs and contribute to performance solutions that scale across systems. Working with software engineers, architects, and AI specialists, you will bring up new training workloads, profile bottlenecks, and build tools to automate setup, debugging, and optimization using cutting-edge

Qualifications

• PhD in Computer Science/Engineering/AI or related field.
• Strong programming in Python and C++.
• Experience with DL frameworks: PyTorch/TensorFlow/JAX.

Responsibilities

• Develop and optimize large-scale AI training workloads on AMD GPUs.
• Profile performance bottlenecks across GPUs, CPUs, memory, and networks.
• Build tooling to automate training setup, debugging, and optimization using AI techniques.
• Collaborate with engineers to evaluate models and training strategies.
• Transform experiments into reusable workflows and best practices.

Skills

Python

C++

PyTorch

JAX

TensorFlow

vLLM

SGLang

Education

PhD in Computer Science/Engineering/AI

Tools

PyTorch Profiler

ROCm Profiler

Nsight

Triton

Job description

ADVANCE YOUR CAREER. ADVANCE THE WORLD.

At AMD, we believetechnology has the power to solve the world’s most important challenges. From advancing healthcare and scientific discovery to powering AI and the technologies people rely on every day, innovation at AMDis shapingthefuture.

Whetheryou'redesigning next-gen processors, enabling AI breakthroughs, orbringing leading edge products to market, every role at AMD contributes to something bigger— technologythat moves the world forward.Join us and, together, we'll advance your career.

As an AMD intern and co-op, you'll be placed at the epicenter of the AI ecosystem, working alongside experts and industry pioneers. You'll do important work, learn new skills, expand your network, and gain real-world experience on projects that impact millions of end-users worldwide. Whether you're an undergrad or a PhD student, your contributions matter—and your experience here will be a launchpad for what comes next.

JOB DETAILS:

• Location: San Jose, CA or Santa Clara,CA
• Onsite/Hybrid: This role requires the student to work full time (40 hours a week), in either a hybrid or onsite work structure throughout the duration of the co-op/intern term
• Duration:
  • Spring/Summer Co-op: January 25, 2027 - August 13, 2027
  • Summer Internship:
    • Semester Students: May 24, 2027 - August 13, 2027
    • Quarter Students: June 21, 2027 - September 10, 2027
  • Summer/Fall Co-op:
    • Semester Students: May 24, 2027 - December 10, 2027
    • Quarter Students: June 21, 2027 - December 10, 2027

WHAT YOU WILL BE DOING:

We are seeking a highly motivated AI Training Systems & Performance Engineering PhD Intern/Co-Op to join our team. In this role, you will help accelerate the adoption and optimization of cutting-edge AI training workloads on AMD Instinct™ GPUs while contributing to the next generation of AI software performance solutions.

You will work alongside software engineers, architects, and AI specialists to bring up new training workloads, analyze performance bottlenecks, and develop innovative tooling that improves scalability, efficiency, and developer productivity.

• We will involve you in developing and optimizing large-scale AI training and fine-tuning workloads running on AMD GPU platforms.
• You will help bring up newly released foundation models and training frameworks, adapting implementations and training recipes for AMD hardware while establishing reproducible correctness and performance baselines.
• We will work with you to profile and analyze AI workloads, identifying bottlenecks across GPUs, CPUs, memory systems, networking, and communication infrastructure.
• You will develop tools and workflows that automate training setup, debugging, performance analysis, and optimization using LLM-powered agents and agentic AI techniques.
• You will investigate and implement optimization strategies that improve training throughput, GPU utilization, memory efficiency, and scalability across distributed multi-GPU environments.
• We will expose you to advanced profiling and performance analysis tools to benchmark and tune AI frameworks, libraries, SDKs, and applications running on AMD platforms.
• You will collaborate with software engineers and architects to evaluate emerging AI models, distributed training techniques, and performance optimization opportunities.
• Your work will help transform successful experiments into reusable workflows, best practices, and software improvements that enhance out-of-the-box AI training performance on AMD hardware.

WHO WE ARE LOOKING FOR:

• Currently pursuing a PhD in Computer Science, Computer Engineering, Electrical Engineering, Artificial Intelligence, Machine Learning, or a related technical discipline.
• Strong programming experience in Python and/or C++.
• Hands-on experience implementing, training, and debugging deep learning models using frameworks such as PyTorch, JAX, TensorFlow, vLLM, or SGLang.
• Experience with one or more of the following areas:
  • Distributed training systems
  • Data, tensor, pipeline, expert, or context parallelism
  • GPU performance optimization
  • AI systems software
  • High-performance computing (HPC)
  • Large language model training and fine-tuning
  • Agentic AI or LLM-powered automation
• Understanding of transformer-based architectures, mixture-of-experts models, and modern LLM training techniques.
• Experience profiling workloads using performance analysis tools such as PyTorch Profiler, ROCm Profiler, VTune, Nsight, or similar tools.
• Familiarity with distributed training technologies and communication libraries such as MPI, NCCL/RCCL, OpenMP, or related frameworks.
• Understanding of GPU architecture, memory systems, communication bottlenecks, and performance tuning methodologies.
• Experience identifying and resolving compute, memory, data-loading, or communication bottlenecks in large-scale AI workloads is preferred.
• Experience with ROCm, HIP, Triton, GPU kernel optimization, or AI systems software development is a plus.
• Publications in AI, Machine Learning, High Performance Computing, Computer Architecture, or related research areas are a plus.

This role is not eligible for visa sponsorship.

Benefits offered are described: AMD benefits at a glance.

AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third-party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law. We encourage applications from all qualified candidates and will accommodate applicants' needs under the respective laws throughout all stages of the recruitment and selection process.

AMD may use Artificial Intelligence to help screen, assess or select applicants for this position. AMD's "Responsible AI Policy" is available here.

This posting is for an existing vacancy.