kernelpool/Qwen3.8-2.4T-A95B-3bit-UVMAX

Mixed-precision (UVMAX) quantization of Qwen/Qwen3.8-2.4T-A95B, converted with mlx-lm from the bf16 release.

What is UVMAX?

UVMAX assigns bit widths per tensor class instead of quantizing uniformly.

tensor class precision parameters size share
Expert FFN gate/up 2-bit, group 128 1,580B 414 GiB 55.2%
Expert FFN down 3-bit, group 128 790B 299 GiB 39.9%
Attention + gated delta net 6-bit, group 64 40B 30.2 GiB 4.0%
Shared experts 8-bit, group 64 4.6B 4.6 GiB 0.6%
Embeddings, LM head 4-bit, group 64 4.1B 2.1 GiB 0.3%
Routers 8-bit, group 64 0.4B 0.4 GiB <0.1%
Norms, convolutions, gates bf16 / fp32 0.01B <0.1%
total 2.66 bits/weight 2.42T 750 GiB

Quality

Teacher-forced against the bf16 release on identical tokens, 48 windows of 1025 tokens. KLD is KL(bf16 ‖ UVMAX) over the full output distribution and is corpus-specific.

corpus ppl bf16 ppl UVMAX ratio mean KLD median KLD top-1 agreement
Linux kernel C 1.300 1.406 1.08× 0.099 0.0008 94.9%
XNU kernel C 2.122 2.272 1.07× 0.106 0.0061 91.7%
JavaScriptCore C++ 2.059 2.184 1.06× 0.102 0.0031 92.1%
English prose 2.684 3.188 1.19× 0.225 0.0528 84.8%
all 1.976 2.172 1.10× 0.133 0.0062 90.9%

Throughput

2× M3 Ultra (512 GB), tensor parallelism over Thunderbolt RDMA, batch size 1, peak memory per node.

prompt tokens prefill tok/s generation tok/s peak memory
4k 176 14.2 385 GiB
16k 171 13.7 385 GiB
32k 162 13.1 386 GiB
64k 142 12.3 391 GiB
128k 113 10.7 402 GiB

Use with mlx

Requires mlx-lm 0.31.3 or later.

pip install mlx-lm

The model does not fit on one 512 GB machine. Serve it across two with an MLX distributed hostfile:

mlx.launch --backend jaccl --hostfile hosts.json --env MLX_METAL_FAST_SYNCH=1 -- \
  mlx_lm.server --model kernelpool/Qwen3.8-2.4T-A95B-3bit-UVMAX

Sampling follows the base model: temperature 1.0, top-p 0.95, top-k 20.

Downloads last month
1,044
Safetensors
Model size
221B params
Tensor type
BF16
·
U32
·
MLX
Hardware compatibility
Log In to add your hardware

2-bit

Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for kernelpool/Qwen3.8-2.4T-A95B-3bit-UVMAX

Quantized
(26)
this model