LLM Reinforcement Learning Framework Engineer

As a key player in the AI revolution, NVIDIA is pushing the boundaries of what's possible in accelerated computing. We are seeking an exceptionally dedicated LLM Reinforcement Learning Algorithm Engineer to join our ambitious team in China. This role is vital in advancing our large language model (LLM) capabilities, particularly in improving reasoning abilities for math, coding, and agentic AI. Join us and contribute to groundbreaking innovations that will craft the future of computing!

What you'll be doing:

• Developing and introducing groundbreaking reinforcement learning algorithms tailored for LLM applications.

• Collaborating with a world-class team of engineers and researchers to integrate these algorithms into applied scenarios.

• Using your extensive expertise in math and AI to improve the reasoning capabilities of our models.

• Engaging in rigorous testing and refinement processes to ensure flawless performance and reliability.

• Contributing to our collective goal of delivering industry-leading AI solutions, strictly adhering to NVIDIA's high standards.

What we need to see:

• Proficient in C++/Python programming.

• 3+ years working experience.

• BS or MS (or equivalent experience) in CS, CE, EE, or a related field.

• Proven experience in reinforcement learning and its application to large language models.

• Strong background in mathematics and AI algorithms, with a focus on reinforcement learning.

• Demonstrated history of applying reinforcement learning algorithms in practical scenarios.

• Understanding of GPU architecture is a huge plus.

• Excellent problem-solving skills and the ability to work collaboratively in a dynamic team environment.

• A passion for innovation and a dedication to achieving outstanding results.