AI Research Infrastructure – Reinforcement Learning Post-Training Intern

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: This role requires the student to work full time (40 hours a week), 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. This role is not eligible for visa sponsorship.

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.