AxionML Gemma-4-31B-NVFP4

Mirrored by AxionML for open-source serving and deployment use cases. Part of AxionML's effort to provide ready-to-serve quantized models for the community.

Quantized by NVIDIA. The weights in this repository are an unmodified copy of nvidia/Gemma-4-31B-IT-NVFP4 (revision 4135a98a9b728a548947683219633b25682223ac). All credit for the quantization belongs to NVIDIA.

This is an NVFP4-quantized version of google/gemma-4-31B-it (30.7B parameters), quantized with NVIDIA Model Optimizer.

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. On Blackwell Tensor Cores, native FP4 multipliers exploit E2M1 simplicity while higher-precision FP32 accumulation protects dot-product accuracy.

Ready for commercial and non-commercial use under Apache 2.0 (Gemma).

Model Summary

Evaluation Results

Scores reported by NVIDIA for this checkpoint. temperature=1.0, top_p=0.95, max_new_tokens=131072.

Quantization Details

• Quantization format: NVFP4 (W4A4, group size 16) on the MLP / feed-forward linear layers; attention, embeddings, lm_head and the vision tower kept in BF16
• KV cache: FP8
• Calibration dataset: cnn_dailymail
• Tool: NVIDIA Model Optimizer

Usage

Deploy with SGLang

python3 -m sglang.launch_server \
    --model-path AxionML/Gemma-4-31B-NVFP4 \
    --quantization modelopt_fp4 \
    --tp 1 \
    --reasoning-parser gemma4 \
    --tool-call-parser gemma4

Deploy with vLLM

vllm serve AxionML/Gemma-4-31B-NVFP4 \
    --quantization modelopt \
    --tensor-parallel-size 8

Sister checkpoint: AxionML/Gemma-4-12B-NVFP4 uses the same MLP-only recipe.

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 and the upstream quantized model card for full details.

Credits

• Base model: google/gemma-4-31B-it
• Quantization: nvidia/Gemma-4-31B-IT-NVFP4 by NVIDIA
• Mirror: AxionML