LLM Reinforcement Learning Framework 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