NVIDIA
From NVIDIA's careers page
Status: listed on the employer's own careers page when we last checked (daily). Confirm on their site before applying.
By joining our Architecture group, you will be part of a dynamic team that is spearheading the AI revolution.
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.
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.
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
A free account lets you apply, save this job, see how well you match and check your resume against it.
Scores your resume against this job the way resume scanners do: the job's hard and soft skills, searchability, recruiter tips and formatting.
Create a free account to see your score. Your resume stays in this browser until then, and we never store it.
Already a member? Sign in