Summer 2027 PhD AI Research Infrastructure RL PostTraining Intern

AMD; Santa Clara (CA); Hybrid; USD 55,000 - 69,000; Full time

Job summary

AMD invites PhD students to join a Summer 2027 internship in Santa Clara, CA, to develop RL-based post-training infrastructure for large language and multimodal models. You will work on scalable training pipelines, multi-GPU systems, and production-quality tooling, with researchers and industry leaders.

You will gain real-world experience, collaborate across teams, and contribute to reproducible research and publications, in a hybrid onsite setting at AMD's Santa Clara campus.

Qualifications

• Pursuing a PhD in a related field with strong programming skills.
• Experience with RL, LLM post-training, RLHF/RLAIF, or preference optimization.
• Familiarity with distributed training, multi-GPU workloads, and large-scale inference.

Responsibilities

• Develop and optimize infrastructure for RL-based post-training of large language and multimodal models.
• Build scalable systems for rollout generation, inference, reward computation, and policy updates.
• Improve distributed training efficiency, fault tolerance, and resource utilization.
• Design interfaces for researchers to implement RL algorithms quickly.
• Build tools for experiment configuration, logging, monitoring, and reproducibility.
• Profile training pipelines and resolve performance and memory bottlenecks.
• Support on-policy and off-policy workflows with verified or model-generated feedback.
• Collaborate with researchers to translate requirements into production-quality infrastructure.
• Document system designs and contribute to technical reports and publications.

Skills

Python

PyTorch

Reinforcement learning

Distributed training

Multi-GPU

Education

PhD in Computer Science or ML/AI/CE

Tools

Docker

Kubernetes

Job description

ADVANCE YOUR CAREER. ADVANCE THE WORLD.

At AMD, we believe technology 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 AMD is shaping the future.

Whether you're designing next-gen processors, enabling AI breakthroughs, or bringing leading edge products to market, every role at AMD contributes to something bigger - technology that moves the world forward. Join us and, together, we'll advance your career.

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.

JOB DETAILS:

• Location: Santa Clara, CA, USA
• 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: Summer 2027 Internship
  • Semester Schools: May 24, 2027 - August 13, 2027
  • Quarter Schools: June 21, 2027 - September 10, 2027

WHAT YOU WILL BE DOING:

• Develop and optimize infrastructure for RL-based post-training of large language and multimodal models.
• Build scalable systems for rollout generation, inference, reward computation, and policy updates.
• Improve distributed training efficiency, reliability, fault tolerance, and resource utilization.
• Design interfaces that enable researchers to implement and evaluate new RL algorithms quickly.
• Build tools for experiment configuration, checkpointing, logging, monitoring, and reproducibility.
• Profile end-to-end training pipelines and resolve performance, memory, and communication bottlenecks.
• Support on-policy and off-policy training workflows using verifiable, preference-based, or model-generated feedback.
• Collaborate with researchers to translate experimental requirements into production-quality infrastructure.
• Document system designs and contribute to technical reports and publications.

WHO WE ARE LOOKING FOR:

• Must be currently pursuing a PhD in Computer Science, Machine Learning, Artificial Intelligence, Computer Engineering, or a related field.
• Strong programming skills in Python and experience with PyTorch.
• Knowledge of reinforcement learning, LLM post-training, RLHF/RLAIF, or preference optimization.
• Experience with distributed training, multi-GPU workloads, or large-scale inference.
• Familiarity with parallelism strategies such as data, tensor, pipeline, or sequence parallelism.
• Experience with training and inference frameworks, orchestration systems, or cluster environments.
• Understanding of GPU performance, memory management, networking, and distributed communication.
• Experience building reliable research infrastructure, profiling systems, or debugging distributed workloads.
• Familiarity with containerization, experiment tracking, and cloud or cluster computing is beneficial.
• Publications at leading venues such as ICML, NeurIPS, ICLR, MLSys, CVPR, ICCV, or ECCV are preferred.

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.