Modal Research and HeavyBall's MRP brings multi-token prediction to diffusion LMs

Modal Research and NYU Shanghai's HeavyBall Research introduce Multi-token Residual Prediction (MRP), a 3-layer module attached to a frozen diffusion LM backbone that predicts the inter-step logit residual instead of the full next-step distribution. On SDAR-4B GSM8K, MRP holds 70.9% accuracy at K=3 versus 5.9% for naive MTP distillation, and in SGLang it reaches up to 1.56x throughput losslessly in speculative mode or recovers up to +16 accuracy points under aggressive low-threshold dynamic decoding. The paper, code, SGLang implementation, and SDAR-MRP models on Hugging Face are released.