Cosmos Lite v0.2.0

Cosmos Lite v0.2.0

This release adds end-to-end quantized deployment support for the 4B NVIDIA
Cosmos3 Edge DROID robot policy on a single RTX 4090, alongside the existing
Cosmos3 Nano and RoboCasa365 pipelines.

Highlights

• Packed W4A16, W8A16, and fixed mixed-precision Marlin inference for
  Cosmos3 Edge.
• Streaming export and self-contained Edge bundles with the native processor,
  vision encoder, VAE, provenance, sizes, and SHA256 integrity metadata.
• Direct bundle validation, serving, replay, latency profiling, and RoboLab
  closed-loop evaluation.
• Four prebuilt Edge DROID variants with model cards, rollout videos, and
  reproducible RTX 4090 benchmarks.
• A validated two-step sampler for Edge, selected from paired closed-loop
  comparisons rather than latency alone.

Recommended Configurations

Environment and task	Quantization	Denoise	VRAM	Request latency	Success rate
RoboLab Banana	W8A16	2	8.71 GB	576 ms	72%
RoboLab Banana	W4A16-GenW8	2	8.79 GB	570 ms	80%

Both rows use guidance 3. Latency and VRAM are RTX 4090 measurements; request
latency excludes RoboLab and IsaacSim environment work. Success rates use 50
paired rollouts. See the benchmark document for the complete four-model
matrix, BF16 references, statistical tests, and protocol details.

Models

Prebuilt Cosmos3 Nano and Edge DROID policy bundles are available in the
Cosmos Lite Hugging Face Collection.

Quick Start

git clone https://github.com/xxxxyu/cosmos-lite.git
cd cosmos-lite
CUDA_VISIBLE_DEVICES=0 examples/quantized_robot_policy/setup.sh

• Cosmos3 Edge benchmark
• RoboLab pipeline
• RoboCasa365 pipeline
• Runtime guide

Cosmos Lite is community maintained and is not an NVIDIA product. NVIDIA
Cosmos Framework remains subject to its upstream license and notices; model
weights may have additional terms at their download locations.