cosmos3-codebase-nav git:20260803.520ada9A

Immutable. This exact content is served forever at /api/v1/blob/52cfb49359848a66.

---
name: cosmos3-codebase-nav
description: Use when navigating the Cosmos3 integration in NPA or locating upstream Cosmos3 framework files, defaults, scripts, configs, recipes, and docs.
---

# Cosmos3 Codebase Navigation

## Source And Attribution

Adapted from NVIDIA cosmos-framework
`skills/atomic/cosmos3-codebase-nav/SKILL.md`.

Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. Used under OpenMDW-1.1.
See `skills/LICENSE-NVIDIA-COSMOS3-OPENMDW-1.1` and
`skills/NOTICE-NVIDIA-COSMOS3`.

## When To Use

Use this skill when the user asks where Cosmos3 setup, access checks, inference
defaults, parser settings, workflows, or post-training configs live. Also use it
when reviewing or modifying PRs that touch Cosmos3 support in NPA.

## NPA Integration Map

Primary NPA files:

| Need | File |
| --- | --- |
| Cosmos3 access, fetch, redaction, parser config, inference arg builder | `npa/src/npa/workbench/cosmos/cosmos3.py` |
| Real Cosmos 3 generation runner (CLI/SDK/YAML share it) | `npa/src/npa/workbench/cosmos/generate.py` |
| Containerized runtime (framework pinned, no weights baked) | `npa/docker/workbench/cosmos3/Dockerfile` |
| Build-time proof the generate deps resolve for every mode | `npa/docker/workbench/cosmos3/verify_env.py` |
| Generate CLI / SDK | `npa/src/npa/cli/workbench/cosmos3.py`, `npa/src/npa/sdk/workbench/cosmos3.py` |
| Generate workflow | `npa/workflows/workbench/npa-workflows/cosmos3-generate.yaml` |
| Human CLI commands for real workflows | `npa/src/npa/cli/cosmos/__init__.py` |
| SDK compatibility exports | `npa/src/npa/workbench/cosmos/__init__.py`, `npa/src/npa/sdk/workbench/cosmos.py` |
| Fetch workflow | `npa/workflows/workbench/npa-workflows/cosmos-fetch.yaml` |
| Text-to-image H100 smoke workflow | `npa/workflows/workbench/npa-workflows/cosmos3-text-to-image.yaml` |
| Unit tests for access, fetch, inference YAML, and skill artifacts | `npa/tests/workbench/test_cosmos3_access.py` |
| CLI tests for `check` and `fetch` | `npa/tests/cli/test_cosmos3_cli.py` |

The agent-skill artifacts live in this repository under:

- `skills/atomic/cosmos3-setup/SKILL.md`
- `skills/atomic/cosmos3-codebase-nav/SKILL.md`
- `skills/atomic/cosmos3-env-troubleshoot/SKILL.md`
- `skills/workflows/cosmos3-inference/SKILL.md`
- `skills/workflows/cosmos3-post-training/SKILL.md`

## What Not To Look For

There is no supported Cosmos skill-display subcommand. Cosmos3 skills are not
platform commands, job templates, or SDK callables. They are SKILL.md files for
coding agents to read.

The retained Cosmos3 workflow YAMLs are real executable specs:

- `npa-workflows/cosmos-fetch.yaml`
- `npa-workflows/cosmos3-generate.yaml`
- `npa-workflows/cosmos3-text-to-image.yaml`

Do not restore `cosmos3-setup.yaml`, `cosmos3-codebase-nav.yaml`,
`cosmos3-env-troubleshoot.yaml`, `cosmos3-post-training.yaml`, or retired raw
SkyPilot task templates as skill launchers unless the product direction changes
to a real executable workflow.

## Upstream Cosmos Framework Map

When a task needs upstream package internals, inspect a clone of
`https://github.com/NVIDIA/cosmos-framework.git`.

Common upstream paths:

| Need | Upstream path |
| --- | --- |
| Sampling args, setup args, override models | `cosmos_framework/inference/args.py` |
| Per-modality defaults | `cosmos_framework/inference/defaults/<mode>/sample_args.json` |
| Inference script | `cosmos_framework/scripts/inference.py` |
| Training script | `cosmos_framework/scripts/train.py` |
| Ray serving presets | `cosmos_framework/inference/ray/configs/latency.yaml`, `throughput.yaml` |
| SFT recipe TOMLs | `examples/toml/sft_config/<recipe>.toml` |
| SFT launch shells | `examples/launch_sft_<recipe>.sh` |
| SFT schema | `cosmos_framework/configs/toml_config/sft_config.py` |
| Setup docs | `docs/setup.md` |
| Inference docs | `docs/inference.md` |
| Training docs | `docs/training.md` |

## Navigation Routine

1. Start with `rg` in NPA:

   ```bash
   rg -n "NPA_COSMOS3|Cosmos3|cosmos3-text-to-image|guardrails" npa
   ```

2. If the answer depends on upstream behavior, inspect the upstream file rather
   than guessing from NPA wrappers.

3. Keep NPA and upstream responsibilities separate. NPA owns CLI/SDK wrappers,
   SkyPilot workflows, image defaults, and tests. NVIDIA's repo owns Cosmos3
   framework semantics, inference arguments, recipes, and training internals.

4. When adding docs or tests, reference SKILL.md artifacts as repository files,
   not as commands.