Summer 2027 PhD ML Systems Research Engineering Intern

Advanced Micro Devices, Inc•Santa Clara, CA

8d•Hybrid

About The Position

As an AMD intern, 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.

Requirements

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.
This role is not eligible for visa sponsorship.

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.