NVIDIA is seeking an engineer for its RL Frameworks team to develop open-source tools and infrastructure for reinforcement learning post-training. The role focuses on architecting and building scalable RL infrastructure that operates efficiently across various hardware. Candidates will work closely with researchers and contribute to improving RL frameworks and distributed runtimes. This position requires a blend of software engineering and collaboration with multiple teams to enhance AI capabilities.

• MS or PhD in Computer Science, Computer Engineering, or a related field or equivalent experience
• 5+ years of professional experience in distributed systems, high-performance computing, deep learning infrastructure, or ML systems engineering
• Strong proficiency in Python and C/C++
• Demonstrated experience building or contributing to large-scale distributed systems or runtime frameworks in production
• Strong verbal and written communication skills and the ability to collaborate across organizational and geographic boundaries
• Depth in reinforcement learning for LLM post-training, PyTorch internals, Kubernetes runtime internals, or end-to-end distributed systems design

• Architect and build RL post-training infrastructure that scales from experimentation on a single GPU to production across thousands of nodes
• Tune RL training-inference-rollout loops on GPUs, CPUs, and LPUs for performance
• Contribute to and improve the performance and usability of open-source RL frameworks
• Partner with teams building CPU-driven rollout workloads and supply the necessary systems and framework engineering
• Advocate for researcher and partner needs with NVIDIA's networking, math library, and compiler teams
• Work with hardware teams to leverage next-generation hardware capabilities in post-training workloads

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.