Agent RL Infra Engineer

Details

We’re hiring an engineer to help us bring reinforcement learning to every agent team at NVIDIA. This is a rare chance to shape how autonomous, self-improving agents learn and evolve across the enterprise. The role sits at the intersection of ML research and production engineering. The challenge is to evaluate emerging approaches, adapt them into enterprise-ready blueprints, and make them available inside sandboxed execution environments with the security and governance the enterprise demands.

Responsibilities

• Split work between creating enterprise-ready RL capabilities and partnering with agent teams to put them into practice.
• Build RL cookbooks and environments:
  • Evaluate and adapt democratized RL approaches into reusable cookbooks and blueprints so agent developers can integrate self-improvement loops (GRPO, DPO, PPO, RLAIF) on their own.
  • Design verifiable reward environments building on NeMo Gym, extending to domain-specific environments for internal use cases.
  • Operationalize NVIDIA and third-party training backends as production services inside Sandbox.
  • Integrate with NeMo Microservices (Curator, Customizer, Evaluator, Guardrails) to enable end-to-end data flywheel workflows for RL.
• Infrastructure, reliability, and collaboration:
  • Lead data curation and active learning strategies to continuously improve training data quality.
  • Design RL training loops for agent self-improvement: reward modeling, policy optimization, safety constraints.
  • Integrate with AI Factory GPU infrastructure for throughput, data locality, and multi-node training.
  • Build observability for training runs and ensure workloads meet security and governance requirements.
  • Collaborate with platform, security, agent infrastructure, and internal customer teams on safe deployment of training outputs.

Requirements

• MS in CS, ML, or related field (or equivalent experience).
• 10+ years of experience.
• Experience operationalizing fine-tuning methods (LoRA, SFT) and especially RL techniques (DPO, GRPO, PPO, RLAIF) into reusable cookbooks and self-service workflows.
• Familiarity with distributed training frameworks (e.g., Megatron, NeMo, DeepSpeed, FSDP, HF Accelerate) and ML ops skills covering pipeline automation, job orchestration, and GPU cluster management.
• Proficiency in Python, Go, Rust, or similar.
• Background in CS, ML, or related field through formal education or equivalent experience.

Ways to stand out

• Building RL environments or training recipes that other teams consumed as self-service capabilities.
• Familiarity with NVIDIA infrastructure (DGX, AI Factory, NVLink/InfiniBand), NeMo Microservices, or the evolving RL-for-agents ecosystem (rLLM, Agent Lightning, HUD, OpenRLHF, SkyRL).
• Experience with data curation, active learning, continuous learning loops, or data flywheel architectures.

Compensation & Benefits

• Base salary range: 224,000 USD - 356,500 USD.
• Eligible for equity and benefits (link provided in original posting).

Other information

• Applications accepted at least until April 2, 2026.
• This posting is for an existing vacancy.
• NVIDIA uses AI tools in its recruiting processes.
• NVIDIA is an equal opportunity employer committed to diversity.