LLM Reinforcement Learning Framework Intern - 2027
At a glance
- Location
- Shanghai, China
- Work type
- Onsite
- Category
- Data
Internship in LLM Reinforcement Learning Framework at NVIDIA in Shanghai, China. Work on Nemo RL, apply state-of-the-art RL algorithms, and improve RL rollout performance within NVIDIA's architecture group. Preferred experience includes VeRL, Miles, Slime, Nemo RL, PyTorch, RLHF, and LLM training frameworks like FSDP/Megatron LLM; pursuing CS/engineering degree.
What they are looking for
Location: China, Shanghai
We are now looking for exceptionally dedicated individuals to join our team as LLM Reinforcement Learning Framework Intern!
At NVIDIA, we are renowned for our continual progress in the fields of graphics and parallel computing. By joining our Architecture group, you will be part of a dynamic team that is spearheading the AI revolution. This remarkable opportunity will allow you to work on innovative projects that extend the frontiers of GPU technology and AI.
What you'll be doing
Help NVIDIA Nemo RL to be efficiently working under large scale post train workload.
Adopt SOTA RL algorithms to Nemo RL.
Improve RL rollout performance.
What we need to see
Having experience with VeRL, Miles, Slime, or Nemo RL is highly beneficial.
Strong proficiency in programming languages such as Python, C++, or CUDA.
Ambitious approach with a proven track record of taking initiative and delivering results.
Experience with PyTorch, RLHF (Reinforcement Learning with Human Feedback), and LLM training frameworks like FSDP/Megatron LLM is a plus.
Familiarity with agentic RL and inference engine development is preferred.
Pursuing A degree in Computer Science, Engineering, or a related field