GLM-5.2-JANGTQ-mxfp4attn

bearzi/GLM-5.2-JANGTQ (2-bit routed experts) with its fp16 attention, shared experts, embeddings and lm_head replaced by the 4-bit mxfp4 tensors from mlx-community/GLM-5.2-mxfp4.

Why

The JANGTQ recipe quantizes routed experts aggressively but keeps attention and shared experts at higher precision, since they are read on every token and carry model coherence. In this source bundle they were left at fp16 rather than the 8-bit affine the recipe specifies - roughly 10% of parameters, read in full each token, dominating decode time.

Swapping them for 4-bit mxfp4 cuts that traffic ~4x. The same change applied to GLM-5.2-JANGTQ_K took decode from 10.6 to 18.8 tok/s on an M3 Ultra; this variant has not been separately benchmarked.

Recipe

tensors format
routed experts JANGTQ 2-bit (unchanged from source)
attention (q_a, q_b, kv_a, o_proj, MLA absorb, indexer) mxfp4 4-bit
shared experts mxfp4 4-bit
embed_tokens, lm_head mxfp4 4-bit
norms, router gate fp16

Note mxfp4 ships MLA attention in absorbed form (embed_q / unembed_out), so the source bundle's kv_b_proj tensors are superseded and unreferenced. Attention at 4 bits is below the 8-bit affine the JANGTQ recipe calls for.

Assembly

Built by server-side copy on the Hub - the source shards were duplicated, the mxfp4-bearing shards copied in from bearzi/GLM-5.2-JANGTQ_K-mxfp4attn, and a new model.safetensors.index.json written mapping each tensor to the right shard. Some shards therefore contain tensors the index does not reference (the superseded fp16 attention). A loader that follows the index rather than globbing model-*.safetensors is required.

Requirements

Stock oMLX cannot load JANGTQ glm_moe_dsa checkpoints. Use:

https://github.com/yohann-bearzi/omlx      branch feat/turboquant-steel-moe
https://github.com/yohann-bearzi/jangq     branch glm52-jangtq

built with OMLX_WITH_CUSTOM_KERNEL=1 (Metal toolchain, Python 3.11-3.13).

Validation

Not benchmarked or evaluated. The equivalent change on GLM-5.2-JANGTQ_K was spot-checked for coherence; this bundle has not been. Treat as untested.

Downloads last month
17
Safetensors
Model size
66B params
Tensor type
F16
·
U32
·
U8
·
BF16
·
MLX
Hardware compatibility
Log In to add your hardware

4-bit

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

Model tree for bearzi/GLM-5.2-JANGTQ-mxfp4attn

Base model

zai-org/GLM-5.2
Quantized
(1)
this model

Collection including bearzi/GLM-5.2-JANGTQ-mxfp4attn