Agent Rl Infra Engineer

Nvidia Corporation

CA, United States
Base: 224,000 usd - 356,500 usd; bonus/equity: equ...
On-site
Reinforcement learning techniques (grpo, dpo, ppo, rlaif)
Distributed training frameworks (megatron, nemo, deepspeed)
Ml ops skills (pipeline automation, job orchestration)
Nvidia Corporation is seeking an experienced engineer to integrate reinforcement learning (RL) capabilities into their agent systems. The role emphasizes operationalizing RL techniques and creating reusable frameworks that empower agent developers

Job Summary

  • We're hiring an engineer to help us bring reinforcement learning to every agent team at NVIDIA, shaping how autonomous, self-improving agents learn and evolve across the enterprise.
  • The role sits at the intersection of ML research and production engineering, evaluating emerging approaches, adapting them into enterprise-ready blueprints, and making them available inside sandboxed execution environments.
  • The work splits between creating enterprise-ready RL capabilities and partnering with agent teams to put them into practice, including building RL cookbooks and environments, and ensuring infrastructure, reliability, and collaboration.

Matching Summary

Match Score: 85

Nvidia Corporation is seeking an experienced engineer to integrate reinforcement learning (RL) capabilities into their agent systems. The role emphasizes operationalizing RL techniques and creating reusable frameworks that empower agent developers.

Salary

Base: 224,000 USD - 356,500 USD; Bonus/Equity: Equity; Benefits: Benefits

Skills & Requirements

Must-have

  • Reinforcement learning techniques (GRPO, DPO, PPO, RLAIF)
  • Distributed training frameworks (Megatron, NeMo, DeepSpeed)
  • ML ops skills (pipeline automation, job orchestration)
  • Proficiency in Python, Go, or Rust
  • Operationalize fine-tuning methods (LoRA, SFT)

Nice-to-have

  • Building RL environments for self-service
  • Familiarity with NVIDIA infrastructure
  • Experience with data curation and active learning

Key Requirements

  • MS in CS, ML, or related field (or equivalent experience)
  • 10+ years of experience

Work Rights

Not specified

Tailored Resume

Cover Letter