nemo-gym-recipe-integration

Integrates NVIDIA NeMo Gym environments into Relax as three-step RL training recipes.

585|144|Updated Apr 13, 2026
One-click install
npx skills add https://github.com/redai-infra/Relax --skill nemo-gym-recipe-integration
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: nemo-gym-recipe-integration
Source: https://github.com/redai-infra/Relax/tree/main/skills/nemo-gym-recipe-integration
Command: npx skills add https://github.com/redai-infra/Relax --skill nemo-gym-recipe-integration

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

Adding a new NVIDIA NeMo Gym environment to the Relax RL framework involves data preparation, local Gym service setup, remote Ray training launch, and verifier validation, with many subtle failure modes. This Skill encodes the proven three-step recipe workflow and accumulated pitfall records so each integration is executable, validated, and reproducible.

Core Features & Use Cases

  • Three-Step Recipe Workflow: Generates the seven required recipe files (env.sh, prepare script, Gym launcher, training script, verifier, README, PITFAIL.md) covering data preparation, local Gym service, and direct Ray training submission.
  • Validation Ladder: Enforces staged checks from static linting through data integrity, service readiness, network reachability, single-trial verification, and multi-sample training with reward variance and optimizer steps.
  • Failure Triage: Maps symptoms like Gateway 404 callbacks, reward-always-zero, and cleanup_unverified to concrete first checks, separating genuine model errors from integration bugs.
  • Use Case: When adding a new agentic RL environment (e.g., a stateful tool-use benchmark) to Relax, use this Skill to produce the complete recipe, validate the verifier distinguishes correct and incorrect responses, and confirm a real training run reaches an optimizer step and checkpoint.

Quick Start

Integrate the specified NeMo Gym environment into Relax as a new recipe under examples/nemo_gym_agentic/recipes, following the three-step prepare, serve, and train workflow.

Frequently Asked Questions about nemo-gym-recipe-integration

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I integrate a NeMo Gym environment into Relax for RL training?

Follow the three-step recipe: prepare the raw JSONL data to a shared path, start a local Docker-based Gym service with its own private Ray cluster, then submit a direct Ray training job. Create the seven required recipe files and validate each stage before proceeding.

How do I validate a NeMo Gym recipe before training?

Use the six-level validation ladder: static checks, data integrity, service readiness via /readyz, network reachability between Gym and Relax, one full trial with expected reward, and a multi-sample training run reaching an optimizer step and checkpoint.

Why is the reward always zero in my NeMo Gym RL training?

Reward zero is not automatically an integration failure. First run the verifier directly to confirm it distinguishes correct and incorrect responses, then check inline think tags, parser mismatches, lost metadata, and whether the evaluator actually executed.

Can I use one rollout sample to verify GRPO training works?

No, one sample only validates plumbing. Use at least four samples per prompt so GRPO computes a meaningful group advantage, and confirm reward variance and a non-zero advantage before declaring success.

Why does the Gateway return 404 or 410 on model callbacks?

Check that the opaque rollout ID prefix is preserved across subprocess and Ray boundaries, the callback URL uses the correct /agentic_api route, and no stale container or old image is serving the environment.