JetBrains ships Mellum2.1 Thinking MTP head: 1.4–1.8x faster decoding with ~90% draft acceptance

pmttyji · reddit · 2026-11-02

JetBrains released an MTP (multi-token prediction) variant of Mellum2.1-12B-A2.5B-Thinking on Hugging Face. The MTP head — one extra transformer block trained jointly with the model — drafts the next token during inference; every drafted token is verified by the main model, so output stays identical while decoding runs 1.4–1.8x faster with 90% of drafts accepted.

Weights live in model-mtp.safetensors under the mtp. prefix, announced via numnextnpredictlayers: 1 in config.json. Usage:

Original post →

More from Infra

Infra channel →