EchoLabs33/mamba-130m-hxq
Mamba-130M-HXQ
3.8x smaller. Pure SSM. Architecture proof. Mamba-130M compressed from 489 MB (FP32) to 128 MB — a pure state-space model proving the codec works beyond transformers. No calibration data. No architecture-specific tuning. Justpip installandfrom_pretrained().
Install and Run
pip install "helix-substrate[hf]"import helix_substrate # registers the HXQ quantizer with HuggingFace
from transformers import AutoModelForCausalLM, AutoTokenizer
model = AutoModelForCausalLM.from_pretrained("EchoLabs33/mamba-130m-helix")
tokenizer = AutoTokenizer.from_pretrained("EchoLabs33/mamba-130m-helix")
inputs = tokenizer("The future of artificial intelligence", return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=64)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))That's it. import helix_substrate registers the quantizer. from_pretrained() handles the rest automatically.
Benchmark
Eval: WikiText-2 test split, 2048 tokens, stride 512.
Good to Know
- +18.4% PPL delta — higher than transformer models. Expected: Mamba-130M is tiny (24 layers, 768 hidden), so each weight carries more information per parameter. The companion Zamba2-1.2B (which includes Mamba2 layers) compresses at +2.90% — SSM architectures compress well at scale.
- GPU and CPU supported — runs on any CUDA GPU or CPU via standard PyTorch. Fused kernels for additional speedup are in progress.
- Fine-tunable via LoRA — compressed weights remain frozen, but LoRA adapters attach to each
HelixLinearlayer viaHelixLinearSTE. Seehelix-substratefor training infrastructure. - Requires `helix-substrate` — the quantizer is not built into transformers. You need
pip install "helix-substrate[hf]". - `mamba-ssm` recommended — without it, falls back to a slower sequential code path.
Why This Model Exists
This is the architecture proof, not the fidelity champion. HelixCode compresses any nn.Linear — including the inproj, outproj, xproj, and dtproj layers inside Mamba's selective scan blocks. No architecture-specific tuning was needed.
What is HelixCode?
HelixCode is a universal weight compression codec based on vector quantization:
- Each weight matrix is replaced by a 256-entry codebook (float32) + uint8 index matrix + optional sidecar corrections for outlier values
- The compressed form is the executable —
HelixLinearperformscodebook[indices] @ xdirectly, no decompression step - Works on any
nn.Linearregardless of architecture (Transformer, Mamba, MLP, CNN) - No calibration data required — unlike GPTQ/AWQ, codebooks are fit from the weights alone
How It Works
import helix_substrateregisters thehxqquantizer with HuggingFacefrom_pretrained()readsquantization_config.quant_method = "hxq"fromconfig.json- The quantizer replaces 96
nn.Linearmodules withHelixLinearshells before weight loading - Safetensors populates the codebook, indices, and sidecar buffers directly
- The model runs in compressed form — no decompression needed
Mamba-Specific Details
Mamba's architecture includes non-weight parameters that are stored at full precision:
A_log— log-space diagonal state matrix (24 layers)D— skip connection parameter (24 layers)dt_bias— timestep bias (24 layers)conv1d— causal convolution (24 layers)
These are not nn.Linear and are not compressed. Only the projection matrices (inproj, outproj, xproj, dtproj) are VQ-compressed.
Compression Receipt
Compressed tensors: 97
From original model: 145 (A_log, D, dt_bias, conv1d, norms)
Total keys: 573
Output size: 128 MB
HXQ ratio: 3.92x (weight bytes)
HelixLinear ratio: 5.61x (in-memory, includes format overhead reduction)
PPL delta: +18.4% (24.60 vs 20.77 dense)
Eval: WikiText-2 test, 2048 tokens, stride=512Companion Models
Same codec, same pip install, multiple architectures:
Citation
@software{helix_substrate_2026,
title={Helix Substrate: Universal Weight Compression via HelixCode},
author={EchoLabs},
year={2026},
url={https://github.com/echo313unfolding/helix-substrate}
}License
Apache 2.0 (inherited from state-spaces/mamba-130m-hf).
