Summer 2027 PhD AI Research Infrastructure, RL Post-Training Intern

Advanced Micro Devices; Santa Clara (CA); Hybrid; USD 50,000 - 67,000; Full time

Job summary

Advanced Micro Devices in Santa Clara, CA is seeking a passionate AI Research intern who is currently pursuing a PhD in CS/ML/AI. You will work full time in a hybrid or onsite setup this Summer 2027, developing RL-based post-training infra for large models and building scalable tools for experimentation and deployment.

You should have strong Python and PyTorch skills, plus experience with RL, distributed training, and multi-GPU workloads.

Qualifications

• Pursuing a PhD in CS/ML/AI or related field.
• Strong Python and PyTorch programming skills.
• Experience with RL, RLHF/RLAIF or preference optimization.
• Experience with distributed training and large-scale inference.

Responsibilities

• Develop and optimize RL-based post-training infrastructure for LLMs and multimodal models.
• Build scalable rollout generation, inference, reward computation and policy updates.
• Improve distributed training efficiency, reliability and memory usage.
• Design interfaces for researchers to implement RL algorithms quickly.
• Create tools for experiment config, logging, checkpointing and monitoring.
• Profile end-to-end training pipelines and resolve bottlenecks.
• Support on-policy and off-policy training workflows with verifiable feedback.
• Document system designs and contribute to technical reports.

Skills

Python

PyTorch

Reinforcement learning

RLHF/RLAIF

Distributed training

Education

PhD in Computer Science or related field

Tools

Experiment tracking

Containerization

Cluster environments

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.

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:

We are seeking highly motivated AI Research intern to join our team. In this role –

• 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 does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services. 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.