DeepSeek-V4.1-Flash-MLX-mixed-4_8bit-engram6

DeepSeek-V4.1-Flash-MLX-mixed-4_8bit-engram6

MLX (Apple Silicon) build of DeepSeek-V4.1-Flash — 754.6B parameters: 40 layers x 384 routed experts, MLA with cross-layer KV-cache sharing (4 compressor-owning layers serve all 40), a two-layer engram hashed n-gram embedding whose tables alone are 196.6B parameters (~40% of the checkpoint), staggered Sinkhorn hyper-connections, and per-layer attention sinks — quantized to 4-bit experts / 8-bit attention&shared / 8-bit engram.

These files are modified: dequantized from the FP8/FP4 release (bit-exact decode of the 32x32-block ue8m0 fp8 and per-32 fp4 packing) and re-quantized; the architecture is unchanged. The 3 multi-token-prediction layers (DSpark markov/confidence heads) are not included; the vision tower and aligner are carried unmodified but the runtime is text-only.

Runtime

deepseek_v41 exists in no runtime — not transformers, not mlx-lm, not mlx-vlm. This checkpoint loads only through the port:

git clone https://github.com/PipeNetwork/deepseek-v41-mlx && cd deepseek-v41-mlx && pip install -r requirements.txt
python scripts/smoke_generate.py /path/to/DeepSeek-V4.1-Flash-MLX-mixed-4_8bit-engram6

from deepseek_v41_mlx.load import load
model, tokenizer = load("/path/to/DeepSeek-V4.1-Flash-MLX-mixed-4_8bit-engram6")

The port was validated against DeepSeek's own inference/model.py (the only reference): fp32 tiny-config parity 1e-6 across prefill / cached decode / chunked prefill, the three QAT fake-quant ops bit-exact, and negative controls proving the fragile paths are load-bearing (rope-inverse 0.84, attention sinks 0.65, cross-layer sharing 0.56 logit shift when broken). Strict loading reports zero missing / zero unexpected tensors. One reference decode bug was found and documented (odd-step indexer reads the wrong layer's keys; 0.67 logit shift — the port uses the owner's cache): see docs/upstream-notes.md in the repo.

Size and what is quantized

476.8 GB on disk. RAM: 1 TB-class machine (477 GB build; a lazy forward transiently needs ~2x).

Quality

Per-layer divergence ladder vs the bf16-dequantized reference — every one of the 40 decoder layers run on identical inputs (16,384 tokens of wikitext-2), teacher-forced and free-running, with the ladder's arithmetic asserted bit-identical to this converter's:

The engram finding that shapes this set: 6-bit engram is indistinguishable from the shipped fp8 tables (free-running 0.1945 vs 0.1948) while 4-bit engram costs +7.3% free-running — but the engram-6 build is 477 GB and a 512 GiB machine cannot run it, so the engram-4 build is the one that fits and the engram-6 build serves 1 TB machines.

Perplexity is not measurable for this build on our 512 GiB machine (the 477 GB build plus activations exceeds it in every load mode — four were tried). Its quality case is the ladder above: its engram treatment is indistinguishable from the shipped fp8 tables, and every other module is identical to the measured engram-4 build (ppl 2.8963 [2.7103, 3.0933]). Strict-loaded: zero missing / zero unexpected tensors.

License

MIT, as the upstream model. Port code: https://github.com/PipeNetwork/deepseek-v41-mlx.