Add NVFP4 model card for Qwen3.5-2B-Base-NVFP4

README.md

@@ -0,0 +1,117 @@
+---
+pipeline_tag: image-text-to-text
+base_model:
+- Qwen/Qwen3.5-2B-Base
+license: apache-2.0
+library_name: transformers
+tags:
+- AxionML
+- ModelOpt
+- Qwen3.5
+- quantized
+- NVFP4
+- nvfp4
+- sglang
+---
+
+# AxionML Qwen3.5-2B-Base-NVFP4
+
+> Developed by [AxionML](https://huggingface.co/AxionML) for open-source serving and deployment use cases. Part of AxionML's effort to provide ready-to-serve quantized models for the community.
+
+This is an NVFP4-quantized version of [Qwen/Qwen3.5-2B-Base](https://huggingface.co/Qwen/Qwen3.5-2B-Base) (2B parameters), quantized using [NVIDIA TensorRT Model Optimizer](https://github.com/NVIDIA/TensorRT-Model-Optimizer). Weights and activations of linear layers are quantized to FP4, reducing disk size and GPU memory by ~4x compared to BF16.
+
+**About NVFP4 quantization:** NVFP4 on Blackwell couples a compact E2M1 FP4 codebook with blockwise FP8 (E4M3) scaling over 16-element micro-blocks, so that 4-bit stored values remain numerically useful for neural-network computation. The E2M1 codebook provides a small, nonuniform set of representable magnitudes up to ±6 and relies on saturating behavior rather than IEEE NaN/Inf encodings to maximize usable range per bit. Using an FP8 block scale (rather than power-of-two-only E8M0) enables fractional scales and error-minimizing scale selection strategies such as dual-pass evaluation comparing "map max to 6" versus "map max to 4 with clipping." On Blackwell Tensor Cores, native FP4 multipliers exploit E2M1 simplicity to reduce multiplier area while higher-precision FP32 accumulation protects dot-product accuracy.
+
+> **Ready for commercial and non-commercial use under [Apache 2.0](https://www.apache.org/licenses/LICENSE-2.0).**
+
+Over recent months, we have intensified our focus on developing foundation models that deliver exceptional utility and performance. Qwen3.5 represents a significant leap forward, integrating breakthroughs in multimodal learning, architectural efficiency, reinforcement learning scale, and global accessibility to empower developers and enterprises with unprecedented capability and efficiency.
+
+## Qwen3.5 Highlights
+
+Qwen3.5 features the following enhancement:
+
+- **Unified Vision-Language Foundation**: Early fusion training on multimodal tokens achieves cross-generational parity with Qwen3 and outperforms Qwen3-VL models across reasoning, coding, agents, and visual understanding benchmarks.
+
+- **Efficient Hybrid Architecture**: Gated Delta Networks combined with sparse Mixture-of-Experts deliver high-throughput inference with minimal latency and cost overhead.
+
+- **Scalable RL Generalization**: Reinforcement learning scaled across million-agent environments with progressively complex task distributions for robust real-world adaptability.
+
+- **Global Linguistic Coverage**: Expanded support to 201 languages and dialects, enabling inclusive, worldwide deployment with nuanced cultural and regional understanding.
+
+- **Next-Generation Training Infrastructure**: Near-100% multimodal training efficiency compared to text-only training and asynchronous RL frameworks supporting massive-scale agent scaffolds and environment orchestration.
+
+
+For more details, please refer to our blog post [Qwen3.5](https://qwen.ai/blog?id=qwen3.5).
+
+
+## Model Overview
+
+- Type: Causal Language Model with Vision Encoder
+- Training Stage: Pre-training & Post-training
+- Language Model
+    - Number of Parameters: 2B
+    - Hidden Dimension: 2048
+    - Token Embedding: 248320 (Padded)
+    - Number of Layers: 24
+    - Hidden Layout: 6 × (3 × (Gated DeltaNet → FFN) → 1 × (Gated Attention → FFN))
+    - Gated DeltaNet:
+        - Number of Linear Attention Heads: 16 for V and 16 for QK
+        - Head Dimension: 128
+    - Gated Attention:
+        - Number of Attention Heads: 8 for Q and 2 for KV
+        - Head Dimension: 256
+        - Rotary Position Embedding Dimension: 64
+    - Feed Forward Network:
+        - Intermediate Dimension: 6144
+    - LM Output: 248320 (Tied to token embedding)
+    - MTP: trained with multi-steps  
+- Context Length: 262,144 natively and extensible up to 1,010,000 tokens.
+
+### Citation
+
+If you find our work helpful, feel free to give us a cite.
+
+```bibtex
+@misc{qwen3.5,
+    title  = {{Qwen3.5}: Towards Native Multimodal Agents},
+    author = {{Qwen Team}},
+    month  = {February},
+    year   = {2026},
+    url    = {https://qwen.ai/blog?id=qwen3.5}
+}
+```
+
+
+## Quantization Details
+
+This model was quantized by applying NVFP4 to the weights and activations of linear operators within transformer blocks. The KV-cache is not quantized. Vision encoder weights are kept in their original precision.
+
+- **Quantization format:** NVFP4 (MLP-only, MSE calibration)
+- **Calibration dataset:** [Nemotron-Post-Training-Dataset-v2](https://huggingface.co/datasets/nvidia/Nemotron-Post-Training-Dataset-v2)
+- **Quantized checkpoint size:** ~1.5GB
+- **Tool:** [TensorRT Model Optimizer](https://github.com/NVIDIA/TensorRT-Model-Optimizer)
+
+## Usage
+
+### Deploy with SGLang
+
+```bash
+python3 -m sglang.launch_server \
+    --model-path AxionML/Qwen3.5-2B-Base-NVFP4 \
+    --quantization modelopt_fp4 \
+    --tp 1 \
+    --reasoning-parser qwen3
+```
+
+### Reproduce with ModelOpt
+
+```bash
+python3 examples/llm_ptq/hf_ptq.py \
+    --pyt_ckpt_path Qwen/Qwen3.5-2B-Base \
+    --qformat nvfp4_mse \
+    --export_path ./qwen3.5-2b-base-nvfp4
+```
+
+## Limitations
+
+The base model was trained on data that may contain toxic language and societal biases. The quantized model inherits these limitations. It may generate inaccurate, biased, or offensive content. Please refer to the [original model card](https://huggingface.co/Qwen/Qwen3.5-2B-Base) for full details.