Description
At NVIDIA, our employees are passionate about AI, HPC, VISUAL, and GAMING. Our SA team focuses on bringing NVIDIA's new technology into different industries. You will work closely with industry sales, developer relationship managers, and product teams in the hiring position.
Job Responsibilities: Drive research, development, and optimization of Reinforcement Learning algorithms and infrastructure for Large Language Models and multimodal models. Collaborate with internal research and engineering teams to adapt and validate state-of-the-art RL methods on NVIDIA GPU platforms at scale. Improve Reinforcement Learning initiatives and engagements with customers, providing technical guidance on integrating NVIDIA RL technologies into their AI workflows. Develop and maintain reusable toolchains, experiment management workflows, and technical documentation to accelerate both internal and customer-facing projects.
Requirements: MS or PhD in Computer Science, Artificial Intelligence, Mathematics, or related fields, with solid foundations in algorithms and programming. 5+ years of experience (including research) in Reinforcement Learning, Large Language Model training, or multimodal learning. Proficient in PyTorch and familiar with RL training frameworks and workflows. Strong engineering skills with experience in distributed training, task orchestration, or evaluation pipelines. Ability to work independently with minimal day-to-day direction, and willingness to conduct exploratory experiments on frontier problems. Desire to be involved in multiple diverse and innovative projects. Outstanding verbal and written communication skills.
Nice to Have: Experience with RLHF, GRPO, DPO, or other alignment and post-training methods for LLMs. Experience with scale-out HPC or cloud architectures for large-scale model training. CUDA optimization or GPU performance tuning experience. Experience with agentic AI systems, code generation models, or multimodal RL. Publications in top-tier venues in RL, NLP, or multimodal learning.