LLM Reinforcement Learning Framework Intern - 2027
At a glance
- Location
- Shanghai, China
- Work type
- Onsite
- Category
- Data
NVIDIA is hiring an LLM Reinforcement Learning Framework Intern in Shanghai, China. You will work on NVIDIA Nemo RL, adopting state-of-the-art RL algorithms, and improving RL rollout performance. The role emphasizes hands-on work with RL, LLM training frameworks, and related toolchains, with a preference for experience in Nemo RL, VeRL, Miles, Slime, PyTorch, and FSDP/Megatron LLM.
What they are looking for
Location: China, Shanghai Department: Intern
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