QDA-Gemma-4-31B-W4A16

4-bit weight quantization of google/gemma-4-31B-it. 2.4× smaller than BF16 (25.9 GB vs. 62.5 GB), on par with BF16 accuracy.

Evaluation

Benchmark BF16
62.5 GB
QDA W4A16
25.9 GB
Google QAT W4A16
23.3 GB
ParoQuant
19.3 GB
AIME 202584.084.482.881.8
HMMT Feb 202677.178.574.075.3
GPQA-Diamond85.785.383.384.4
IFBench80.480.279.180.4
MMLU-Pro86.986.486.386.2
LiveCodeBench83.985.183.183.4
Average83.083.381.481.9

Thinking mode, pass@1 averaged over 1–32 seeds. Same sampling, token caps and seeds for every model; the ParoQuant column ran on its own serving stack (vLLM 0.19.1 with a config override it needs to load, and the chat template shipped in its repo, which differs from Google's current template by one newline in the system turn).

Speed on one H100 80GB, 512 generated tokens per request (tok/s = generated tokens / wall time, mean of 2 runs): 95 tok/s at batch 1 and 1,101 to 1,251 at batch 16 (the two runs); ParoQuant 66 and 891 (both on vLLM 0.19.1).

Usage

Runs on NVIDIA H100, A100, RTX Pro 6000 and B200 GPUs.

vLLM:

vllm serve qda-lab/QDA-Gemma-4-31B-W4A16

SGLang:

python -m sglang.launch_server --model-path qda-lab/QDA-Gemma-4-31B-W4A16

How it was made

Weights are 4-bit integers in groups of 32, quantized with GPTQ; the group scales were then fine-tuned by distillation. The vision tower stays in BF16. The recipe was chosen by QDA's automated search.

Reproduce

hf download qda-lab/qda-recipe-inputs --repo-type dataset --include "QDA-Gemma-4-31B-W4A16/*" --local-dir inputs
qda build inputs/QDA-Gemma-4-31B-W4A16/gemma-4-31b-w4a16.yaml --set "devices=[0,1]"

Needs the QDA library (the qda command). Recipe: gemma-4-31b-w4a16.yaml; inputs (calibration sample and distillation data): QDA-Gemma-4-31B-W4A16/ in qda-lab/qda-recipe-inputs. Takes about 1 h on 2× H100 (measured). The rebuild matches this checkpoint's KL to BF16 within 5 %; weights are not bit-identical.

License

Apache-2.0, the same as the base model (Gemma 4 license).

Citation

TBD.

Downloads last month
71
Safetensors
Model size
31B params
Tensor type
BF16
·
I32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for qda-lab/QDA-Gemma-4-31B-W4A16

Quantized
(363)
this model