PhD ML Systems Research Intern — AI Infra & Scale

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

Job summary

Advanced Micro Devices, Inc. (AMD) in Santa Clara, CA, invites PhD students for a 40-hour per week intern/co-op focused on ML systems. Collaborate with researchers to build scalable AI infrastructure, from distributed training to model serving, and improve efficiency across ML workflows.

The role emphasizes Python and PyTorch experience, strong problem-solving skills, and the ability to translate experimental requirements into production-ready infrastructure.

Qualifications

• Must be currently enrolled in a PhD program in Computer Science, Computer Engineering, Electrical Engineering, Artificial Intelligence, or a related field.
• Programming experience in Python.
• Coursework or hands-on experience in artificial intelligence, or computer systems.
• Familiarity with PyTorch or similar machine learning frameworks.
• Interest in distributed systems, reinforcement learning, LLMs, or AI infrastructure.
• Strong analytical, problem-solving, and communication skills.

Responsibilities

• Develop and optimize infrastructure for distributed training, reinforcement learning, inference, and model optimization.
• Build scalable systems for experiment management, rollout generation, model serving, and evaluation.
• Improve training and inference efficiency through parallelism, scheduling, checkpointing, caching, and performance optimization.
• Develop tools that integrate AI models with engineering systems, evaluation pipelines, and benchmarking frameworks.
• Support experiment tracking, logging, monitoring, and reproducibility across ML workflows.
• Analyze system performance and help identify improvements in reliability, scalability, and resource utilization.
• Collaborate with researchers and engineers to translate experimental requirements into production-quality infrastructure.
• Document system designs, experimental results, and technical implementations.

Skills

Python

PyTorch

Distributed systems

Reinforcement Learning

ML Infrastructure

Education

PhD in CS/EE/CE/AI

Job description

Advanced Micro Devices, Inc. (AMD) in Santa Clara, CA, invites PhD students for a 40-hour per week intern/co-op focused on ML systems. Collaborate with researchers to build scalable AI infrastructure, from distributed training to model serving, and improve efficiency across ML workflows.

The role emphasizes Python and PyTorch experience, strong problem-solving skills, and the ability to translate experimental requirements into production-ready infrastructure.