CLI Commands
This page documents the NeMo Gym command-line interface.
All functionality is exposed through a single gym entry point, organized into command groups (gym <group> <command>). ng is a drop-in alias for gym. Every group and command supports -h/--help.
The legacy ng_* / nemo_gym_* commands (such as ng_run or nemo_gym_collect_rollouts) still work but are deprecated. Each one prints a notice pointing at its gym replacement and then runs it. See Migrating from the legacy commands at the bottom of this page.
Quick Reference
Common Options
These options are shared across many commands.
The --benchmark, --environment, --resources-server, --model-type, and --agent-type selectors resolve a component name to its config file for you, so you do not need to know the project’s directory layout. If you mistype a name, the CLI suggests the closest match. To point at a config file directly, use --config <path> instead.
Selecting the model server
Commands that need a model (gym env start, gym eval run) configure it through four flags:
--model-api-key passed on the command line appears in shell history. Prefer setting policy_api_key in your env.yaml instead.
Swapping the agent
--agent-type NAME runs an environment or benchmark with a different agent harness, without editing any
config. It resolves to responses_api_agents/NAME/configs/NAME.yaml; use NAME/FLAVOR to pick a
non-default config in that directory.
Available on gym env start, gym env validate, gym env prefetch, and gym eval run. Like the other
selectors it is pure shorthand, so --agent-type terminus_2_sandboxed_agent and --config <that file>
behave identically.
Run gym list agents to see the harnesses you can pass.
The composed server instance is renamed after the agent that runs it, so metrics and gym env status
report the harness actually used: composing terminus_2_sandboxed_agent onto
terminal_bench_2_1_opencode_sandboxed_agent produces terminal_bench_2_1_terminus_2_sandboxed_agent.
gym env start prints the instance list, which matters for the two-step flow:
--agent-type chooses the harness to compose; --agent names the running instance to collect rollouts
from. Because the harness is settled once the servers are up, --agent-type is rejected with
--no-serve.
Anything that routes rows — --agent, +agent_map values, +fan_out entries — must name the composed
instance, and Gym says so rather than failing later:
Rows collected before the swap are re-routed for you, so existing rollout files stay usable.
Some tasks only work with the harness they were built for. A resources server may pin the harnesses
required for the task with allowed_agents, and
Gym refuses a pairing it does not list, before any server starts:
Results from an undeclared pairing might not be valid.
Hydra overrides (escape hatch)
The gym CLI is a thin wrapper over Gym’s Hydra config system. Standard flags (--flag value) cover the common inputs. Anything not covered by a flag can still be passed as a raw Hydra override using +key=value (add a new key) or ++key=value (add or override an existing key). Unknown overrides are forwarded to Hydra untouched, so advanced config composition is fully functional.
When a command needs overrides for keys that have no dedicated flag, you can keep the whole command in Hydra form (config paths included) or mix --flag and +key=value styles in one invocation:
General
gym --help
List all command groups and their descriptions.
gym --version
Print the NeMo Gym version along with Python, key dependency, and system information.
Discovery
Commands for discovering benchmarks, environments, model servers, agent harnesses, and resources servers.
gym list <type> lists all components of that type; gym list <type> <name> inspects that one.
All commands accept:
--jsonwhich prints machine-readable output instead of a human-readable table.--search-dirwhich allows you to surface your own Gym-compatible components alongside the built-ins.
gym list benchmarks
List the benchmarks available in NeMo Gym, with their kind, manifest status, domain, and description.
The value in the Name column is the benchmark identifier and can be passed to --benchmark.
To see which agent harness a benchmark composes with, inspect it with gym list benchmarks <name>.
gym list environments
List the manifest-first catalog unioned with legacy environment and benchmark configs. Manifest-backed entries expose their declared metadata; unmigrated entries remain visible with no-manifest status.
The JSON output includes three path fields:
path: the entry’s directory. Multiple entries can share this directory, so it is not a unique identifier.config_path: the entry’s config file. Use this field to locate the config; it is not necessarily insidepath.manifest_path: the entry’smanifest.yaml, ornullwhen the entry has no manifest.
All non-null paths are absolute with symlinks resolved. A path reached through a symlink therefore uses its target’s location in the output.
For example, the config for aalcr/config_v1_1 is benchmarks/aalcr/config_v1_1.yaml, beside its entry directory benchmarks/aalcr/config_v1_1/.
gym list agents
List the agent harnesses under responses_api_agents/, with each one’s composition pattern — composable (Pattern A: references a separate resources server, so it can be wired into a matching environment) vs self-contained (Pattern B: ships its own framework/environment) — plus its config variants.
gym list models
List the model servers under responses_api_models/.
The value in the Model column corresponds to a config flavor and can be passed as the --model-type flag to other commands.
gym list resources-servers
List the resources servers under resources_servers/.
The value in the Name column corresponds to the config flavor and can be passed as --resources-server.
gym search
Filter a listing to fuzzy matches on a query (against each component’s name/value, description, domain, and type-specific fields). Without a type, the command searches the unified environment and benchmark catalog.
Datasets
Commands for preparing, previewing, and managing datasets. By default dataset transfer commands use HuggingFace; pass --storage gitlab to target the GitLab Registry.
HuggingFace commands require an HF_TOKEN environment variable (or huggingface-cli login). GitLab commands authenticate via mlflow_tracking_token in your YAML config (or the MLFLOW_TRACKING_TOKEN environment variable). See Downloading from HuggingFace for HF authentication details.
gym dataset init
Scaffold a dataset in the Harbor task format: dataset.toml with a [gym] table and one task with an
instruction, a verifier and a reference solution. Refer to Harbor Tasks.
gym dataset pull
Fetch a Harbor dataset into environments/harbor/ of the Gym checkout, or under the working directory when Gym is
installed as a package ($NEMO_GYM_DATASETS_DIR when set), and print where it landed, its task count and its
dataset digest. Nothing is prepared, built or run. The fetch is the
one gym eval run and gym dataset validate make for a harbor: target, so a folder that was edited by hand,
left by an interrupted fetch of another version, or written for another package is refused in the same words;
a second pull of an unchanged dataset downloads nothing and says so. Refer to
Harbor Tasks.
gym dataset validate
Run each Harbor task’s reference solution (solution/solve.sh) through its verifier with the oracle harness
and print one line per task with its reward, or why there is none. gym eval run TARGET --agent harbor_oracle
is the same run. Results go under results/harbor/validation/<taskset>/ (rollouts.jsonl unless --output
names another file, the generated run config and skipped.json), apart from any evaluation of the same target,
so an evaluation never resumes from the oracle’s rewards. The exit code is 0 when every selected solution scored
1, and 1 when a solution scored below 1, was masked or failed, when a task folder did not load, or when no
selected task had a solution to run. Refer to Harbor Tasks.
gym dataset upload
Upload a prepared local JSONL dataset to HuggingFace (default) or GitLab.
gym dataset download
Download a dataset from HuggingFace (default) or GitLab.
gym dataset rm
Delete a dataset from the GitLab Registry. Prompts for confirmation.
gym dataset migrate
Migrate a JSONL dataset to HuggingFace from GitLab. Use gym dataset upload if you do not want automatic GitLab deletion.
gym dataset collate
Validate and collate a dataset, generating metrics and statistics.
gym dataset render
Generate a dataset preview by materializing prompts from a raw input file and a prompt template, producing JSONL with populated responses_create_params.input for RL training.
Each input row must not already have a populated responses_create_params.input; the command applies the prompt template from --prompt-config to each row, fills in the input, and preserves the row’s other fields.
Which data-preparation command should I use?
gym dataset render— a focused, standalone step that applies a prompt template to raw rows to populateresponses_create_params.input. No servers are started. Use it when you have raw data and just need to turn it into prompt-ready rows.gym dataset collate— the full preparation pipeline for training: it can download missing datasets, validate data, and compute dataset metrics, writing train/validation splits and metrics artifacts. Use it to prepare and validate datasets for training or PR submission.
Environments
Commands for developing, running, and inspecting environments. An environment defines tasks, verification, and its interaction surface; Gym config composes it with agent, model, and runtime components.
gym env init
Scaffold a resources server, or create a manifest-backed environment or benchmark skeleton. Workload scaffolds include a manifest, Gym config, sample data, and the extension points selected by the integration profile.
gym env resolve
Resolve the configs, flags, and overrides into a final merged config and print it. Useful for debugging configuration. Secrets are hidden.
Unlike gym env validate, resolve substitutes no dummy model, so a model config’s policy_* values must be supplied for its interpolations to resolve.
When an environment server block doesn’t match its server’s config class, resolve still prints the config and lists the fields to fix as a warning on stderr.
gym env validate
Validate a manifest-backed workload or legacy config without starting Ray or server subprocesses. Manifest validation resolves the Gym config offline, checks composition mirrors and datasets, and reports the resolved components. Legacy validation retains the existing config pre-flight behavior, and also checks each environment server block against the config class of the server it runs, which gym env start otherwise applies only when that server launches. Exits 0 when valid, or 1 with a clean message when not.
By default, values a run needs but the config leaves undefined (such as policy_base_url from env.yaml) are filled with placeholders so the rest can still be checked, and reported as a warning. Pass --strict to fail on them instead.
gym env packages
Each server has its own isolated virtual environment. List the packages installed in a server’s environment.
gym env test
Exercise a manifest-backed workload’s verifier fixture in its resources-server environment, or run the existing pytest suite for one or all resources servers.
gym env publish
Run the local publication check for a manifest-backed workload. The command runs static validation and the verifier fixture, rejects manifest metadata placeholders, and confirms that the exact manifest is discoverable. It includes an experimental annotation only when the manifest’s flag is true. It does not certify the workload, assess implementation quality, require a full evaluation or training run, commit, or push changes.
gym env start
Start the NeMo Gym servers (agents, models, resources, and environment servers) defined by the provided configs. Reads configuration from YAML files and runs each configured server in its own environment. Rollout collection dispatches /run through environment servers.
gym env status
Show all currently running NeMo Gym servers and their health.
gym env prefetch
Pre-warm per-server venvs without starting any servers. Accepts the same config flags as gym env start. For each server in the config, creates its isolated venv and installs dependencies serially. Intended for use in Dockerfile builds so venvs are ready at runtime without network access.
gym env schema
Print a resources server’s TaskData schema as JSON Schema. Exits non-zero if the server does not export a task_data.py schema.
Evaluation
Commands for running evaluations end to end: prepare data, collect rollouts, aggregate sharded runs, verify rollout health, and profile results.
gym eval prepare
Prepare a benchmark’s data by running its prepare.py script and dump the result to disk.
gym eval plan
Prepare one benchmark and write its plan: every task once, sorted by task id, with a digest of its prepared row. The plan stores ids and digests, not the inputs themselves. The format is versioned ("schema": "nemo-gym.eval-plan.v1"), only changes additively within a version, and is printed as JSON Schema by --schema.
Each task’s id comes from its row’s task_id, problem_id or instance_id, and its taskset is the dataset’s declared taskset (the benchmark name when none is declared). Unlike gym eval run, there is no fallback to the row’s position, so benchmarks without such a field are refused. The digest covers the prepared row after excluding runtime routing and bookkeeping fields (agent_ref, task_source, skills_ref, and keys beginning with _ng_). A prompt template (prompt_config) is applied when a run starts, so editing the template does not change the digest.
gym eval run
Collate data, start the servers, and collect rollouts. This is the main evaluation command. By default it spins up all required servers. Pass --no-serve to collect against servers you already started with gym env start.
In place of --benchmark, a positional TARGET runs a dataset in the Harbor task format: a task folder, a folder of task folders, a harbor:<dataset>[@<version>] hub reference, or the name of a committed entry under environments/harbor/ (for example terminal_bench_4) after gym dataset pull. Refer to Harbor Tasks for the layout and the --sandbox, --only-tasks and --exclude-tasks options that apply to such a target.
Generation parameters
The most common sampling parameters have dedicated flags on gym eval run: --temperature, --top-p, and --max-output-tokens. These map onto responses_create_params.temperature, responses_create_params.top_p, and responses_create_params.max_output_tokens.
Any other responses_create_params field that has no dedicated flag can be set with a raw Hydra override using the ++responses_create_params.<field> syntax. Overrides are merged into each input row’s existing responses_create_params with a shallow merge (top-level keys only):
Because the merge is shallow, setting a field inside a nested object, such as ++responses_create_params.reasoning.effort=low, replaces the row’s entire nested dictionary at that key. Other fields under the same nested object are not preserved.
Output files
Next to --output (<stem> is its name without .jsonl), the run writes:
Resume interrupted runs
Pass --resume to restart the same command after a crash or interruption and pick up only the rows that have not finished yet.
How it works:
- Materialized inputs. On the first run, the fully expanded input rows (after
--num-repeats,--limit,--prompt-config, and any overrides) are written to a sidecar file next to your output. The path is derived from--outputby appending_materialized_inputsto the stem — sorollouts.jsonlproducesrollouts_materialized_inputs.jsonl. - Incremental output. Successful rollouts are flushed to the main output JSONL after each completion; retryable and terminal failures, plus intentional omissions (
_ng_omitted), are saved in a<stem>_failures.jsonlsidecar, so partial progress survives a crash. - Failed agent calls. By default a failed agent
/runends the run, because the alternative is a score reported over fewer rollouts than were asked for. Before it ends, the run writes one<stem>_failures.jsonlrow for that rollout with its identity, failure class, failure report and_ng_failure_message, marked_ng_omitted_rollout: trueand_ng_unrouted_run_error: true. That row is evidence of the lost reply, not an attempt: no reader scores or counts it, readers skip it before they take a rollout’s last attempt, so an older failure of the rollout still counts. An uncertain or possibly delivered request blocks ordinary resume; proven single-attemptnot_sentfailures remain eligible within the saved budget. Pass+route_failures_to_sidecar=trueto let the run continue instead: the call is then recorded in the same sidecar as one attempt, with no reward. Every rollout that leaves the score is logged as it happens, whichever layer classified it, and the totals appear with the closing artifact paths. The class says whether the rollout ran:agent_run_errorwhen the agent itself answered, since a NeMo Gym agent returns 500 when its handler raises, andagent_request_failedfor a gateway status (429, 502, 503, 504) or no usable reply, neither of which says anything about the rollout. Other rollouts keep running, the score is computed over completed rollouts only, and resume re-dispatches only eligible, classified pre-answer failures up toNEMO_GYM_MAX_ROLLOUT_ATTEMPTS. Saved judge answers and unclassified execution are not regenerated. A run in which nothing produced a result fails instead of reporting an empty score, unless an option below counts its rollouts as zeros. - Counting failures in the score. Metrics are computed over completed rollouts only.
+count_failure_classes_as_zero=[<class>]includes the named sidecar rows in/aggregate_metricsso they land in the denominator:[agent_run_error]counts the rollouts that reached the agent and broke, which is what a model emitting unparsable tool calls looks like from here, and needs+route_failures_to_sidecar=trueto have anything to count. A row that carries no reward is scored zero for the metric input only, so neither the sidecar nor the rollouts JSONL records a verdict that no verifier gave. It works the same ingym eval runandgym eval aggregate. A sidecar row whose producer marked it_ng_failure_counts_as_zero: trueis counted the same way whatever its class, when it is the rollout’s latest attempt: Simple Agent’soutcome_policy.by_failure_kind.model_policy_response_timeout=count_as_zeroopts in only when every observed attempt of the final operation timed out waiting for a response after a confirmed completed upload;final_on_model_timeout=trueis a compatibility alias. Missing upload or attempt evidence cannot activate this rule. - Counting lost rollouts.
+count_missing_rollouts_as_zero=truescores as zero a rollout that is in the materialized inputs but left no row in the output JSONL or the sidecar, such as one killed mid-flight or onedispatch_budget_snever started. A sidecar row written by+record_omitted_rollouts(see Rollouts with no result) records no result, so such a rollout is counted here. The zero goes into the metric input only, so it does not itself make an interrupted rollout safe to replay.--resumeruns only eligible work; saved answers and uncertain execution still block regeneration. A rollout whose newest attempt has a known failure is never counted here, so this does not widen+count_failure_classes_as_zero. The count is reported ascoverage/imputed. With--disable-aggregationnothing is scored, so a run with no row still fails.gym eval aggregatechecks each shard against its own rows, materialized inputs and sidecar. - Matching. Resume validates the saved manifest against the current dataset, materialized inputs, and task configuration. A configuration that differs only in operational settings (timeouts, retries, ports, concurrency, health limits, paths, tracking) resumes with a warning naming them and a line in
<stem>_manifest.resume-drift.jsonl; a difference in a semantic setting (data and limits, repeats, agent and model, sampling, judge and grading) is refused, naming each setting. See the resume guide for the full split. The run manifest reserves execution identities and selects each logical rollout’s newest attempt. Completed results are reused, including zero rewards; eligible failed or interrupted work receives a new attempt identity. - Terminal failures. A sidecar row stamped
_ng_failure_terminalis never retried.+retry_terminal_timeouts=trueretries rows of classtimeout_exceededeven when they are stamped terminal. Retries still stop atNEMO_GYM_MAX_ROLLOUT_ATTEMPTS. Off by default, because some agents stamp their timeouts terminal on purpose. - Dispatch budget.
+dispatch_budget_s=<seconds>stops starting new rollouts that many seconds after collection starts; set it to the usable walltime of the allocation. It requires--concurrencyor+max_resident_rollout_tasks; without either limit, the whole queue could be sent before the budget is checked. Rows that are not started are not attempts, so the next--resumeruns them.+drain_margin_s=<seconds>also stops starting rows when less than that much time is left; unset, the margin is the p75 duration of the run’s recorded attempts once there are at least 5. An attempt whose agent/runreturns a usable reply is always recorded, even when the agent reports the rollout itself as failed or timed out; an attempt whose/runfails (an error status, or no usable reply) is recorded only if it ran at least as long as the median so far, because a quick failure says nothing about how long rollouts take. Every run ends with a latency summary over the same attempts (median, p90, p99, max) and the number of rows not started. - Rollouts with no result. An attempt the agent reports as
kill_shaped(Slurm SIGTERM, a dead Ray actor, an OOM outside the rollout) is always saved in<stem>_failures.jsonlas an interruption: it is the rollout’s newest attempt, spends no retry, and blocks automatic replay when its execution cause is unknown. An unused failure budget does not establish that replay is safe. A rowdispatch_budget_snever started is not an attempt and by default leaves no row; run coverage counts it asnot_started.+record_omitted_rollouts=truewrites one sidecar row for each such row, so the absence has a record. The row keeps the rollout’stask_indexandrollout_index, and carries_ng_no_persist: true,_ng_omitted_rollout: true,_ng_dispatch_drained: true, the classcancelledand a reason in_ng_failure_message. Such a row is evidence, not an attempt or a failure: the run history skips it, and so do scoring,+count_failure_classes_as_zero, aggregate and exporter metrics, the run health summary,gym dev compare, rollout uploads andgym eval reverify --judge-failed-only. A sidecar written without a run manifest can also holdkill_shapedrows from this option; there an interrupted row is the rollout’s last attempt, so an older failure is not counted again and the rollout is missing. Rows carrying only_ng_no_persist, such as thekill_shapedrows ATIF reverification writes, are read as before. Off by default while readers of the sidecar outside NeMo Gym learn to treat these rows as not run rather than failed. A collector that is itself killed before it sees an outcome leaves no row, and its reserved attempt is counted as missing. A customrollout_collection_driversuch as GDPVal’s multi-stage one keeps its own persistence and ignores the option. - Longest first.
+dispatch_longest_first=truestarts the rows whose earlier failed attempts ran longest first on resume, using theelapsed_secondsthose attempts recorded. Rows with no recorded timing keep their input order and start after them. Without it, rows start in input order.
Changed inputs or semantic configuration (data, limits, repeats, agent, model, sampling, judge and grading) are rejected on resume; operational changes resume with a warning. Legacy runs without a manifest also require an explicit override. +allow_unsafe_resume=true reuses saved inputs when their identity cannot be verified or deliberately differs; it does not bypass corrupt or conflicting history. For a submitted run, gym eval resubmit --recover --allow-unsafe-resume --unsafe-resume-reason "<why>" sets it on the queued job and records the reason and user. Prefer a new output path when evaluating changed tasks or models.
When require_complete=true (the default for gym eval submit), exit status 75 means incomplete with eligible
retries and progress: new successes, omissions, terminal failures or counted failures. Admission timeouts and
nonterminal kill_shaped interruptions are saved but do not establish progress. Retryable work without progress
exits 1 and spends the Slurm infrastructure retry limit. Terminal failures or exhausted
budgets with nothing eligible to retry exit 76; the Slurm chain stops. If at least one task completed and the only
gaps are intentional omissions, collection exits 0 with a count warning, without treating omissions as completed or
scored tasks. A run where every task is omitted exits 76 when completion is required. Local gym eval run and
standalone collection default to require_complete=false and can return 0 with partial results. Use
+require_complete=true to require completion. Explicit Ctrl-C also retains the existing partial-result behavior
when completion is not required; inspect coverage to determine what finished.
See Resume and recover evaluations for artifact requirements, attempt selection, and the current limits on judge-only recovery.
gym eval submit
Submit a benchmark evaluation job to Slurm via a submit config YAML file. Renders job scripts and calls sbatch for each benchmark defined in the config. Use --dry-run to preview the generated scripts without submitting.
gym eval resubmit
Queue a benchmark of an earlier gym eval submit again. Without --recover the job starts with a fresh retry, walltime and per-rollout attempt allowance. With --recover it first decides one action from the run’s saved evidence (resubmit, re-judge saved answers, nothing, or a typed stop) and keeps the spent budgets; see Recover from saved evidence. With --recover it prints one JSON line, and the exit status is 0 only when gate_task.status is SUCCESS, so --inspect exits 1 while work is planned.
gym eval aggregate
Merge sharded rollout results into a single rollouts file with aggregate metrics. Reads every JSONL file matching --input-glob, recomputes aggregate metrics over the global union of records, and writes a <output stem>_aggregate_metrics.json next to the merged rollouts. It also lists the shards in <output stem>_sources.json, which gym eval health-check reads for a merged rollouts.jsonl. The output may be a shard only when it is the only one, reached through its own directory rather than a file link from elsewhere, and then nothing is listed. Use this to combine shards produced by gym eval run --no-serve --disable-aggregation.
gym eval export
Export a finished run with an exporter, chosen by the optional EXPORTER argument. atif is the default; mlflow and wandb send a finished run to those tracking backends, and evalrdl uploads it to EvalRDL. Exporter settings are Hydra +key=value overrides; the option flags below are equivalent forms for atif.
atif exports supported ng_trajectory attachments from a Gym rollouts JSONL file as ATIF v1.7. The command writes one ATIF
trajectory per rollout plus a manifest that preserves Gym’s task and rollout identity. It is a strict offline conversion:
if Gym cannot represent a source trajectory completely in the supported ATIF subset, the export fails instead of silently
omitting data. All invalid JSONL rows are reported together, and no destination is published if any row fails. Captured
model attempts not selected by a turn, such as failed retries, remain in extra.nemo_gym.surplus_model_calls; they do not
become canonical ATIF steps or contribute to final usage totals. The destination directory must not already exist.
This command does not parse ATIF and does not load NeMo Relay. Each ATIF trajectory takes its agent name from the source
row’s agent_ref.name; --agent-version records the version of that agent implementation.
Exporting to MLflow or W&B
mlflow and wandb log what a live run would have logged: the run’s saved config (<stem>_resolved_config.yaml),
its aggregate metrics, and its rollouts. They need <stem>_aggregate_metrics.json (newer than the rollouts file) and the
saved config next to the rollouts, so they cannot export a run that predates the saved config. Backend settings and
credentials (mlflow_tracking_uri, wandb_api_key, …) come from the export command itself (arguments or env.yaml),
never from the run’s saved config, which masks them. A backend SDK’s own environment variables, such as
WANDB_API_KEY, are also used when the command sets no key. Rollouts are uploaded unless the run’s saved
upload_rollouts is false; pass +upload_rollouts=true|false to override. The rollouts file itself is only needed
when the rollouts are uploaded, so +upload_rollouts=false also exports a run whose rollouts file is gone.
Once a run is exported, <stem>_<exporter>_export_done is written next to the rollouts: a version-1 receipt that names
the backend run, the sha256 of each run file it sent, whether the rollouts were uploaded and the rollout projection
version. A later export of the same run is skipped without contacting the backend while the receipt matches the
current files. A receipt that no longer matches them, or a legacy or malformed marker, fails the export and sends
nothing; it is left in place for you to remove deliberately before exporting again. A failed or partial export leaves no
receipt, so it can be retried. A run logged live gets its receipt at live completion when every upload landed and the
artifacts match (an existing receipt is never replaced); its <stem>_<exporter>_run_id alone shows only that the live
run opened, not that every upload landed, so without a receipt the post-run export sends the run again as a separate run
and reports the live run id.
Exporting to EvalRDL
evalrdl uploads one run: the whole directory the rollouts file is in. Everything there is part of the upload (the
rollouts and their sidecars, health reports, model-call captures and any benchmark-specific files), and nothing is copied
or picked out, so keep each run in a directory of its own (the export is refused when the directory holds another
rollouts file, because EvalRDL would read both as one run). The run still needs <stem>_aggregate_metrics.json and
<stem>_resolved_config.yaml there, like the mlflow and wandb exporters. The evalrdl SDK is installed into the
exporter’s own virtual environment, so it never enters Gym’s environment. The SDK is served by the EvalRDL
service’s own package index, <service URL>/v1/client/simple/; evalrdl is not on PyPI. Configure that index through
uv’s own index settings (for example UV_EXTRA_INDEX_URL) when you run the command, and do not point uv at an index that
could carry another package named evalrdl, because the SDK runs with your token in its environment. The SDK reads the
token from $EVALRDL_TOKEN, $GITLAB_TOKEN or $GITLAB_PRIVATE_TOKEN, then from glab, which cluster nodes usually
lack, so set one of the variables there. The environment is built once and the SDK is not upgraded afterwards; when the
service stops accepting an old client, the export fails and names the environment to delete, or pin a newer version with
+requirement=evalrdl==VERSION.
The result is succeeded when EvalRDL stored the run as new, and skipped when it already held the same content. With
+wait=false the run counts as succeeded once EvalRDL has stored it, and the upload token is in the result. When
waiting, an upload EvalRDL stored without a verdict before the timeout fails with exit code 3, and one it refused fails
with exit code 4; both name the reason. A worker still running after +timeout plus an hour for the upload itself is
killed and the export fails.
See ATIF Export for supported trajectories, preserved evidence, and conversion limitations.
gym eval status
Read saved progress for a run directory or a Gym job ID from gym eval submit:
Collection without a batch manifest writes <output-stem>_batch_status.json next to its output JSONL
(for example, gsm8k_batch_status.json), and a shard recorded through run_examples writes
rollouts-<shard>_batch_status.json. Manifest-backed runs retain batch_status.json. A status without a
manifest carries the count, failure and aggregation keys only; the manifest-only keys (recipe, dataset and
metric expectations, repeat policy) belong to batch_status.json.
The directory command lists every stem-derived status file and the legacy batch_status.json when present. Progress is
written at initialization, on the first durable result, at most every five seconds thereafter,
and at completion. Counts report completed and remaining rollouts and the
latest outstanding failures by class and by failure kind, plus the model-final results by kind (an
answered agent deadline, or a failure its producer settled as a scored zero): these are reported as
facts and never counted as completed. A successful retry clears its earlier failure from
these counts. Failed rollouts remain in the remaining count because they have no successful result.
Status starts from what the run’s Store selected (the current attempt of every rollout), never from raw
rollouts.jsonl, so an older success is not counted after a newer retry.
--json prints a mapping from each status-file path or submitted benchmark name to its saved status, plus an
export entry: for each exporter that left a receipt (<stem>_<exporter>_export_done) or a live run id
(<stem>_<exporter>_run_id) beside the output, whether the export completed and the run it logged to. The
human-readable output prints the same as export receipt: mlflow present (run id ...), or missing.
Human-readable output includes the expected count, aggregation status, update time, and tracking state. File targets or path-like
targets that are not directories produce a run-directory error rather than a job-ID lookup.
Gym job IDs use the local submission index; submitted artifacts must be mounted or copied to
the paths recorded in the submission. Missing submitted artifacts are shown as unavailable.
The command reads files without contacting a scheduler or remote host.
For runs without a manifest, status is best-effort: inconsistent resume identities or native result
identities warn once and disable status reporting while collection continues. Gym tries once to write
tracking: "disabled", the exception type as reason, and updated_at, replacing unreliable counts.
Fresh runs clear their output stem’s previous status before initializing the tracker. Manifest-backed runs
retain strict validation. Resume loading retains compact identities rather than full input payloads,
and general-run progress counters update incrementally as results arrive.
gym eval health-check
Verify rollout quality for an existing run directory. The command reads <run-dir>/rollouts.jsonl by default, or the
shards gym eval aggregate merged into it, and writes
quality_summary.json and rollout_verdicts.jsonl in the run directory.
Refer to Rollout Health Checks for verdict semantics, evidence requirements, the complete check catalog, and report schemas.
gym eval profile
Compute a reward profile from collected rollouts. Outputs per-task statistics such as average reward, standard deviation, min/max, and pass rate, useful for filtering tasks before training by difficulty or variance. Requires rollouts collected with --num-repeats greater than 1.
Writes three files next to the rollouts, named from its stem:
See Repeat-Level Metrics for the full field list and how to read the variability statistics.
gym eval compare
Compare a baseline eval run against a candidate run and write a report.
The report contains a table of point-estimate metrics (the change, plus each side’s value and confidence interval), split into key metrics and all other metrics, and a per-task sample-flips table. Existing statistics such as percentile and confidence interval fields do not get separate rows. Each side’s separate intervals stay at 95%: recorded intervals for a single run, recomputed intervals for pooled repeats. For eligible metrics with at least two finite repeat-level values on each side and cross-repeat means, the command also computes a two-sided unpaired Welch confidence interval for candidate minus baseline, at 95% by default. Use --confidence-level to change the delta confidence level; the JSON report records it in the top-level confidence_level field. The command does not issue a pass/fail verdict.
See Compare Results for the evaluation workflow and interpretation of the confidence interval.
Writes into the output directory:
To pool runs, use --baseline-runs b1/rollouts.jsonl,b2/rollouts.jsonl --candidate-runs c1/rollouts.jsonl,c2/rollouts.jsonl,c3/rollouts.jsonl --output-dir results/pooled-comparison. Each repeat contributes one observation to a single Welch comparison; source run means and confidence interval endpoints are not averaged. Either side can use its single-run option. Quality metrics require complete scored repeat estimates in every source; completion-token metrics retain their available observations and report missing accounting. Current single-repeat aggregates have no quality repeat estimates, so collect at least two scored repeats per source before pooling. Whole-run-only metrics cannot be pooled this way. See Pool independent runs for assumptions and examples.
For pooled JSON sides, the ordered sources list contains the complete provenance. Top-level rollout and aggregate paths identify the first source for compatibility. Reports in an existing output directory are replaced.
If a run was collected with --disable-aggregation it has no *_aggregate_metrics.json; run gym eval aggregate first, or supply aggregate files with --baseline-agg-metrics for --baseline, --baseline-runs-agg-metrics for --baseline-runs, and --candidates-agg-metrics for either candidate option.
gym eval compare-suite
Compare candidate and baseline runs across several benchmarks for each model, and across a declared set of model–benchmark pairs. This command normalizes selected metrics, weights benchmarks equally, and reports delta = candidate - baseline with unpaired Welch confidence intervals at a default confidence level of 90%.
Run paths and optional agent or environment-server selectors come from the suite YAML. See Compare Results for a complete multi-model suite YAML, source options, theoretical metric bounds, normalization, equal weighting, statistical assumptions, and coverage/exclusion behavior.
Writes suite_compare_report.md and/or suite_compare_report.json. The suite JSON format is versioned independently with "schema_version": "1" and has no level field. A pooled side’s path and identity fields represent its first source; its ordered sources list contains complete provenance. The separate gym eval compare command keeps its run-level reports and defaults to 95% difference intervals.
gym eval reverify
Recompute rewards by replaying a stored response through a resources server’s /verify endpoint — without re-running model inference. The command accepts native Gym rollouts or the initial completed, text-only Relay ATIF v1.7 profile. It starts the resources server automatically from the provided config.
Before starting, the command checks each resources server’s GET /reverify_mode response. Resources servers report one of three modes:
Servers that report unsupported or unknown cause the command to abort unless --force is passed.
Native Gym input also accepts the rollout collection option +retry_terminal_timeouts=true (with --resume). When a rollout row carries deliverables_dir or reference_ids, both are sent to /verify with the stored response.
ATIF input requires a stateless resources server and rejects --force, --resume, --judge-failed-only, and --append. It fails closed on unsupported or lossy canonical trajectory shapes, including tool-bearing trajectories routed to a resources server with expose_tools_over_mcp: true, and preserves source identity under _ng_atif_provenance without sending that metadata to the verifier. Provider-native payloads and producer-private status records in optional ATIF extra metadata are not interpreted; their producer owns conversion into canonical ATIF fields. A terminal tool step is supported when every call has a correlated observation result.
See Reverify Rollouts for a full walkthrough.
Contributor Helpers
gym dev test
Run NeMo Gym’s core unit tests with coverage reporting.
gym dev compare
Compare two rollout JSONL files task by task. A temporary migration aid: before an old resources server is deleted in favour of its Harbor-path replacement, run both on the same tasks and use this report as the gate.
Rows are joined by task id (_ng_task_id on the Harbor path; task_name or a verifier_metadata id on legacy servers; override with --key FIELD). When both sides span several tasksets, tasks are keyed and listed as taskset/task_id. When only one side does, the other side keeps its plain task ids and each is matched to the task id that appears in exactly one taskset (or in that side’s own taskset); an id that recurs across tasksets is reported as ambiguous and listed as missing. A masked-failures sidecar (<name>_failures.jsonl) is folded back in, keeping only the last record per rollout and none for a rollout that was later scored; a record that names no task is listed as masked under unknown (<side>) #<n>. Each task is classified as identical, flipped (split into old-win and new-win), masked, or missing (only one side has it). A task is masked when either side has mask_sample true, a reward that is absent or null, or a NaN or infinite reward (reported with the raw value). A failure_kind on a row with mask_sample false (such as the Harbor server’s scored zero for a verifier timeout) does not mask: the reward counts, and a flip it causes is listed as a flip with the failure_kind printed beside the side that carries it. A flip where one side also reports an agent error is listed separately as a rerun candidate. A string or boolean reward is an input error that names the file, line and task. When a side has several rows for a task, the mean reward is compared. For every flip the report prints both rewards and the last --tail lines (default 15) of the losing side’s verifier output, read from verifier_logs_dir (resolved against the losing side’s own file directory, then the current directory, then --logs-root) or from the row’s embedded test_output; a log that cannot be read is reported as unavailable with the reason. gym --json dev compare OLD NEW prints the JSON summary on stdout instead of the report (the same object --json PATH writes). The exit code is 0 for a report and 1 when an input cannot be read or the --json PATH file cannot be written.
Migrating from the legacy commands
The legacy ng_* and nemo_gym_* are deprecated and will be removed in the future release.
Use the tables below to find their gym replacement and update your scripts and workflows.
Command mapping
Replacing Hydra overrides with flags
The most common Hydra overrides now have dedicated flags:
For standalone rollout collection, num_samples_in_parallel limits active requests, while max_resident_rollout_tasks limits how many rollout tasks the driver creates at once. When both are set, active requests cannot exceed the lower limit. Both settings are optional; leaving the task window unset preserves the previous admission behavior. Set retain_results_in_memory=false for artifact-oriented runs: results are appended and flushed as they complete, and run_from_config() returns [] instead of the ordered full result list. Aggregation and upload may still load the persisted results into memory. These settings are available through the collection config or Hydra overrides such as +max_resident_rollout_tasks=64 and +retain_results_in_memory=false.
Common workflows, before and after
Run any command with --help to see its full set of flags, and gym --help to list every group.