Arsh9210 commited on
Commit
efd5325
·
verified ·
1 Parent(s): e9c0270

Added MODELCARD.md

Browse files
Files changed (1) hide show
  1. MODELCARD.md +117 -0
MODELCARD.md ADDED
@@ -0,0 +1,117 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Nemotron-Labs-Audex-2B Overview
2
+
3
+ ## Description:
4
+ Nemotron-Labs-Audex-2B is an audio language model built on a dense-2B LLM with strong audio abilities and no degradation on text intelligence. It uses NV-Whisper as audio encoder and outputs both text tokens and audio tokens from XCodec and XCodec-2. The model leverages Cascaded-RL based on Nemotron Cascade 2 for strong text reasoning abilities and operates in both thinking and instruct modes.
5
+ Nemotron-Labs-Audex-2B was developed by NVIDIA as a part of Nemotron.
6
+ _This model is for research and development only._
7
+
8
+
9
+ ### License/Terms of Use:
10
+ NVIDIA OneWay Noncommercial License
11
+
12
+ Audex is derived from outputs of XCodec2 (CC-BY-NC 4.0) and XCodec (Apache-2.0) and is subject to the same non-commercial restrictions. See [`license/THIRD_PARTY_NOTICES.md`](license/THIRD_PARTY_NOTICES.md) for third-party license and attribution details.
13
+
14
+
15
+ ### Deployment Geography:
16
+ Global
17
+
18
+
19
+ ### Use Case:
20
+ Developers building AI agent systems, chatbots, and applications requiring advanced reasoning and long-context understanding in both text and audio domains.
21
+
22
+
23
+ ### Release Date:
24
+ Hugging Face [June 8, 2026] via [HF Model Link](https://huggingface.co/nvidia/Nemotron-Labs-Audex-2B)
25
+
26
+
27
+
28
+ ## Reference(s):
29
+ [Unified Audio Intelligence Without Regressing on Text Intelligence](https://arxiv.org/)
30
+ [UALM: Unified Audio Language Model for Understanding, Generation and Reasoning](https://arxiv.org/abs/2510.12000)
31
+ [Nemotron-Cascade 2: Post-Training LLMs with Cascade RL and Multi-Domain On-Policy Distillation](https://arxiv.org/abs/2603.19220)
32
+
33
+ ## Model Architecture:
34
+ **Architecture Type:** Transformer
35
+ **Network Architecture:** Dense GPT
36
+ **This model was developed based on GPT.**
37
+ **Number of model parameters:** 2B (3.0*10^9)
38
+
39
+
40
+ ## Input:
41
+ **Input Type(s):** Text, Audio
42
+ **Input Format:** String, .wav
43
+ **Input Parameters:** One-Dimensional (1D), One-Dimensional (1D)
44
+ **Other Properties Related to Input:** Text input is provided as prompts or instructions with context length 1M. Audio input uses NV-Whisper as the audio encoder.
45
+
46
+
47
+ ## Output:
48
+ **Output Type(s):** Text, Audio
49
+ **Output Format:** String, Tensor
50
+ **Output Parameters:** One-Dimensional (1D), One-Dimensional (1D)
51
+ **Other Properties Related to Output:** Supports text token generation and audio token generation from XCodec and XCodec-2. Text generation supports 1M context lengths. Audio token generation supports 15 minutes of audio at 16kHz sampling rate.
52
+
53
+
54
+ 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.
55
+
56
+ ## Software Integration:
57
+ **Runtime Engine(s):**
58
+ * vLLM
59
+ * Transformers
60
+
61
+ **Supported Hardware Microarchitecture Compatibility:**
62
+ * NVIDIA Blackwell
63
+ * NVIDIA Hopper
64
+
65
+ **Supported Operating System(s):** Linux
66
+
67
+ The model can be integrated using various checkpoints and APIs: text-only vLLM checkpoint with vLLM inference API, full vLLM checkpoint with vLLM inference API for audio generation, and full HF checkpoint with HF inference API for audio understanding. Conversion scripts are provided.
68
+
69
+ The integration of foundation and fine-tuned models into AI systems requires additional testing using use-case-specific data to ensure safe and effective deployment. Following the V-model methodology, iterative testing and validation at both unit and system levels are essential to mitigate risks, meet technical and functional requirements, and ensure compliance with safety and ethical standards before deployment.
70
+
71
+
72
+ ## Model Version(s):
73
+ Nemotron-Labs-Audex-2B
74
+
75
+ ## Training, Testing, and Evaluation Datasets:
76
+
77
+
78
+ ## Training Dataset:
79
+
80
+ **Data Modality:**
81
+ * Text
82
+ * Audio
83
+
84
+
85
+ **Audio Training Data Size:** 10,000 to 1 Million Hours
86
+ **Text Training Data Size:** 1 Billion to 1 Trillion Tokens
87
+ **Data Collection Method by dataset:** Hybrid: Human/Synthetic
88
+ **Labeling Method by dataset:** Hybrid: Human/Synthetic
89
+ **Properties (Quantity, Dataset Descriptions, Sensor(s)):** Training data is composed of 477.9B tokens in total sourced from public commercial and academic datasets. It includes a mix of Nemotron-Cascade-2 data, Parakeet data, Magpie-TTS data, Audio Flamingo 3 data, and SAO data for various tasks the model supports.
90
+
91
+ ### Testing Dataset:
92
+
93
+ **Data Split:** Not Applicable: All testing benchmarks are separate, standard research benchmarks
94
+ **Data Collection Method by dataset:** Hybrid: Human/Synthetic
95
+ **Labeling Method by dataset:** Hybrid: Human/Synthetic
96
+ **Properties (Quantity, Dataset Descriptions, Sensor(s)):** Not Applicable: All testing benchmarks are separate, standard research benchmarks
97
+
98
+ ### Evaluation Dataset:
99
+
100
+ <img src="assets/result_text.png" alt="Description" style="width: 75%;">
101
+ <br></br>
102
+
103
+
104
+ **Data Collection Method by dataset:** Undisclosed
105
+ **Labeling Method by dataset:** Undisclosed
106
+ **Properties (Quantity, Dataset Descriptions, Sensor(s)):** Evaluated on diverse reasoning benchmarks including mathematical problem solving, code generation challenges, question answering tasks, and instruction following assessments.
107
+
108
+
109
+
110
+ ## Inference:
111
+ **Acceleration Engine:** vLLM
112
+ **Test Hardware:** NVIDIA Hopper (H100)
113
+
114
+ ## Ethical Considerations:
115
+ 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. 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.
116
+
117
+ Please report model quality, risk, security vulnerabilities or NVIDIA AI Concerns [here](https://app.intigriti.com/programs/nvidia/nvidiavdp/detail).