PhD AI Research Infra Intern: RL Post-Training

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

Job summary

AMD Internship Program is seeking a highly motivated AI Research intern in Santa Clara, CA. You will work with senior researchers on RL-based post-training for large language and multimodal models, building scalable, production-grade infrastructure and tools.

You should be pursuing a PhD in Computer Science, ML, or AI, with strong Python and PyTorch skills and knowledge of RL, RLHF/RLAIF, and distributed training. This internship offers hybrid onsite work for Summer 2027.

Qualifications

• Pursuing a PhD in a related field.
• Strong Python and PyTorch skills.
• Experience with RL, LLM post-training, RLHF/RLAIF, or preference optimization.

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, 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.

Skills

Python

PyTorch

Reinforcement learning

Distributed training

LLM post-training

GPU performance

Containerization

Experiment tracking

Cloud computing

Publications in ML venues

Education

PhD in Computer Science / ML / AI or related

Job description

AMD Internship Program is seeking a highly motivated AI Research intern in Santa Clara, CA. You will work with senior researchers on RL-based post-training for large language and multimodal models, building scalable, production-grade infrastructure and tools.

You should be pursuing a PhD in Computer Science, ML, or AI, with strong Python and PyTorch skills and knowledge of RL, RLHF/RLAIF, and distributed training. This internship offers hybrid onsite work for Summer 2027.