Posttrain documentation
Post-training for Cosmos models: offline SFT, DPO, GRPO, actor-critic PPO, on-policy distillation (OPD), and
DiffusionNFT on one substrate that scales from a single GPU to a multinode Ray cluster.
New public users should start with the
general run guide, then read the reference guide for the
algorithm they need. The glossary.md is worth five minutes before anything else — replica,
rank, dp_shard, ActorGroup, and DeviceGroup all mean specific things here.
Getting started
| Document |
What it covers |
howto/run-posttrain.md |
Step-by-step container recipes for GRPO, DPO, PPO, OPD, NFT, and Joint PPO, using a four-GPU showcase that can be adapted |
howto/run-posttrain-qa.md |
Run length and short runs, Joint PPO's silent pre-training evaluation, checkpoint intervals, and first-run errors |
glossary.md |
Terminology, including the three unrelated meanings of "validation" |
Run a capability
Each guide follows the same skeleton, so a reader who has used one can navigate the others.
Customize
Look something up
Qualified commands
Each command below was run end to end in the runtime image of the
run guide and completed successfully. Rerun them after a dependency or
image change.
| Command |
Result |
--validate-manifest for GRPO, PPO, OPD |
exit 0, bind_check: ok |
GRPO GSM8K public recipe (run guide) |
Completed optimizer updates with public Qwen and GSM8K artifacts |
DPO public T2I recipe (run guide) |
Four-rank FSDP initialized from the converted Cosmos3-Nano and completed optimizer updates over synthetic preference pairs |
PPO Qwen3-0.6B public recipe (run guide) |
Completed five policy/critic update iterations |
OPD Countdown public recipe (run guide) |
Completed five disaggregated policy/rollout/teacher updates over NIXL |
NFT Cosmos3-Nano public recipe (run guide) |
Completed five updates against the deterministic toy reward service |
π0.5 Joint PPO (public guide) |
Completed five LIBERO collection/update windows |
GR00T-N1.7 Joint PPO (public guide) |
Completed five LIBERO collection/update windows with exit code 0 |