Cosmos
Safetensors
English
qwen2_5_vl
nvidia

You need to agree to share your contact information to access this model

This repository is publicly accessible, but you have to accept the conditions to access its files and content.

Log in or Sign Up to review the conditions and access this model content.

Cosmos-Reason1: Physical AI Common Sense and Embodied Reasoning Models

Cosmos | Code | Paper | Paper Website

Model Overview

Description:

Cosmos-Reason1 Models: Physical AI models understand physical common sense and generate appropriate embodied decisions in natural language through long chain-of-thought reasoning processes.

The Cosmos-Reason1 models are post-trained with physical common sense and embodied reasoning data with supervised fine-tuning and reinforcement learning. These are Physical AI models that can understand space, time, and fundamental physics, and can serve as planning models to reason about the next steps of an embodied agent.

The models are ready for commercial use.

Model Developer: NVIDIA

Model Versions

The Cosmos-Reason1 includes the following model:

  • Cosmos-Reason1-7B: Given a text prompt and an input video, think and generate the answer with respect to the input text prompt and video.

License:

This model is released under the NVIDIA Open Model License. For a custom license, please contact [email protected].

Under the NVIDIA Open Model License, NVIDIA confirms:

  • Models are commercially usable.
  • You are free to create and distribute Derivative Models.
  • NVIDIA does not claim ownership to any outputs generated using the Models or Derivative Models.

Important Note: If You bypass, disable, reduce the efficacy of, or circumvent any technical limitation, safety guardrail or associated safety guardrail hyperparameter, encryption, security, digital rights management, or authentication mechanism (collectively “Guardrail”) contained in the Model without a substantially similar Guardrail appropriate for your use case, your rights under this Agreement NVIDIA Open Model License Agreement will automatically terminate.

Deployment Geography:

Global

Use Case:

Physical AI: Space, time, fundamental physics understanding and embodied reasoning, encompassing robotics, and autonomous vehicles (AV).

Release Date:

Model Architecture:

Architecture Type: A Multi-modal LLM consists of a Vision Transformer (ViT) for vision encoder and a Dense Transformer model for LLM. Network Architecture: Qwen2.5-VL-7B-Instruct.

Cosmos-Reason-7B is post-trained based on Qwen2.5-VL-7B-Instruct and follows the same model architecture.

Input

Input Type(s): Text+Video/Image

Input Format(s):

  • Text: String
  • Video: mp4
  • Image: jpg

Input Parameters:

  • Text: One-dimensional (1D)
  • Video: Three-dimensional (3D)
  • Image: Two-dimensional (2D)

Other Properties Related to Input:

  • Use FPS=4 for input video to match the training setup.
  • Append Answer the question in the following format: <think>\nyour reasoning\n</think>\n\n<answer>\nyour answer\n</answer>. in the system prompt to encourage long chain-of-thought reasoning response.

Output

Output Type(s): Text

Output Format: String

Output Parameters: Text: One-dimensional (1D)

Other Properties Related to Output:

  • Recommend using 4096 or more output max tokens to avoid truncation of long chain-of-thought response.
  • Our AI models are designed and/or optimized to run on NVIDIA GPU-accelerated systems. By leveraging NVIDIA’s hardware (e.g. GPU cores) and software frameworks (e.g., CUDA libraries), the model achieves faster training and inference times compared to CPU-only solutions.

Software Integration

Runtime Engine(s):

Supported Hardware Microarchitecture Compatibility:

  • NVIDIA Blackwell
  • NVIDIA Hopper

Note: We have only tested doing inference with BF16 precision.

Operating System(s):

  • Linux (We have not tested on other operating systems.)

Usage

See Cosmos-Reason1 for details.

  • Post Training: Cosmos-Reason1 provides examples of supervised fine-tuning and reinforcement learning on embodied reasoning datasets.

Evaluation

Please see our technical paper for detailed evaluations on physical common sense and embodied reasoning. Part of the evaluation datasets are released under Cosmos-Reason1-Benchmark. The embodied reasoning datasets and benchmarks focus on the following areas: robotics (RoboVQA, BridgeDataV2, Agibot, RobFail), ego-centric human demonstration (HoloAssist), and Autonomous Vehicle (AV) driving video data. The AV dataset is collected and annotated by NVIDIA. All datasets go through the data annotation process described in the technical paper to prepare training and evaluation data and annotations.

Data Collection Method:

  • RoboVQA: Hybrid: Automatic/Sensors
  • BridgeDataV2: Automatic/Sensors
  • AgiBot: Automatic/Sensors
  • RoboFail: Automatic/Sensors
  • HoloAssist: Human
  • AV: Automatic/Sensors

Labeling Method:

  • RoboVQA: Hybrid: Human,Automated
  • BridgeDataV2: Hybrid: Human,Automated
  • AgiBot: Hybrid: Human,Automated
  • RoboFail: Hybrid: Human,Automated
  • HoloAssist: Hybrid: Human,Automated
  • AV: Hybrid: Human,Automated

Metrics: We report the model accuracy on the embodied reasoning benchmark introduced in Cosmos-Reason1. The results differ from those presented in Table 9 due to additional training aimed at supporting a broader range of Physical AI tasks beyond the benchmark.

RoboVQA AV BridgeDataV2 Agibot HoloAssist RoboFail Average
Accuracy 87.3 70.8 63.7 48.9 62.7 57.2 65.1

Dataset Format

Modality: Video (mp4) and Text

Dataset Quantification

We release the embodied reasoning data and benchmarks. Each data sample is a pair of video and text. The text annotations include understanding and reasoning annotations described in the Cosmos-Reason1 paper. Each video may have multiple text annotations. The quantity of the video and text pairs is described in the table below. The AV data is currently unavailable and will be uploaded soon!

RoboVQA AV BridgeDataV2 Agibot HoloAssist RoboFail Total Storage Size
SFT Data 1.14m 24.7k 258k 38.9k 273k N/A 300.6GB
RL Data 252 200 240 200 200 N/A 2.6GB
Benchmark Data 110 100 100 100 100 100 1.5GB

We release text annotations for all embodied reasoning datasets and videos for RoboVQA and AV datasets. For other datasets, users may download the source videos from the original data source and find corresponding video sources via the video names. The held-out RoboFail benchmark is released for measuring the generalization capability.

Inference:

Acceleration Engine: PyTorch, flash attention
Test Hardware: H100, A100, GB200

  • Minimum 2 GPU cards, multi nodes require Infiniband / ROCE connection

Ethical Considerations

NVIDIA believes Trustworthy AI is a shared responsibility and we have established policies and practices to enable development for a wide array of AI applications. When downloaded or used in accordance with our terms of service, developers should work with their internal model team to ensure this model meets requirements for the relevant industry and use case and addresses unforeseen product misuse.

Users are responsible for model inputs and outputs. Users are responsible for ensuring safe integration of this model, including implementing guardrails as well as other safety mechanisms, prior to deployment.

For more detailed information on ethical considerations for this model, please see the subcards of Explainability, Bias, Safety & Security, and Privacy below.

Please report security vulnerabilities or NVIDIA AI Concerns here.

Plus Plus (++) Promise

We value you, the datasets, the diversity they represent, and what we have been entrusted with. This model and its associated data have been:

  • Verified to comply with current applicable disclosure laws, regulations, and industry standards.
  • Verified to comply with applicable privacy labeling requirements.
  • Annotated to describe the collector/source (NVIDIA or a third-party).
  • Characterized for technical limitations.
  • Reviewed to ensure proper disclosure is accessible to, maintained for, and in compliance with NVIDIA data subjects and their requests.
  • Reviewed before release.
  • Tagged for known restrictions and potential safety implications.

Bias

Field Response
Participation considerations from adversely impacted groups protected classes in model design and testing: None
Measures taken to mitigate against unwanted bias: None

Explainability

Field Response
Intended Application & Domain: Physical AI Reasoning
Model Type: Transformer
Intended Users: Physical AI developers
Output: Text
Describe how the model works: Generates text answers based on input text prompt and video
Technical Limitations: The model may not follow the video or text input accurately in challenging cases, where the input video shows complex scene composition and temporal dynamics.
Verified to have met prescribed NVIDIA quality standards: Yes
Performance Metrics: Quantitative and Qualitative Evaluation. Cosmos-Reason1 proposes the embodied reasoning benchmark and physical common sense benchmark to evaluate accuracy with visual question answering.
Potential Known Risks: The model's output can generate all forms of texts, including what may be considered toxic, offensive, or indecent.
Licensing: NVIDIA Open Model License

Privacy

Field Response
Generatable or reverse engineerable personal information? None Known
Protected class data used to create this model? None Known
Was consent obtained for any personal data used? None Known
How often is dataset reviewed? Before Release
Is there provenance for all datasets used in training? Yes
Does data labeling (annotation, metadata) comply with privacy laws? Yes
Applicable Privacy Policy NVIDIA Privacy Policy

Safety

Field Response
Model Application(s): Physical AI common sense understanding and embodied reasoning
Describe the life critical impact (if present). None Known
Use Case Restrictions: NVIDIA Open Model License
Model and dataset restrictions: The Principle of least privilege (PoLP) is applied limiting access for dataset generation and model development. Restrictions enforce dataset access during training, and dataset license constraints adhered to. Model checkpoints are made available on Hugging Face, and may become available on cloud providers' model catalog.
Downloads last month
3,118
Safetensors
Model size
8.29B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for nvidia/Cosmos-Reason1-7B

Finetuned
(269)
this model

Datasets used to train nvidia/Cosmos-Reason1-7B

Collection including nvidia/Cosmos-Reason1-7B