nasosger / MuToRLinks
[NeurIPS '25] Multi-Token Prediction Needs Registers
☆24Updated 2 months ago
Alternatives and similar repositories for MuToR
Users that are interested in MuToR are comparing it to the libraries listed below
Sorting:
- The this is the official implementation of "DAPE: Data-Adaptive Positional Encoding for Length Extrapolation"☆39Updated last year
- The official implementation for [NeurIPS2025 Oral] Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink…☆105Updated 2 months ago
- ☆35Updated 8 months ago
- AdaSplash: Adaptive Sparse Flash Attention (aka Flash Entmax Attention)☆29Updated last month
- Official PyTorch implementation and models for paper "Diffusion Beats Autoregressive in Data-Constrained Settings". We find diffusion mod…☆108Updated 3 weeks ago
- Unofficial Implementation of Selective Attention Transformer☆17Updated last year
- ☆26Updated this week
- Remasking Discrete Diffusion Models with Inference-Time Scaling☆54Updated 8 months ago
- PyTorch implementation of StableMask (ICML'24)☆14Updated last year
- ☆106Updated 2 months ago
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆105Updated last month
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆133Updated 2 weeks ago
- DeciMamba: Exploring the Length Extrapolation Potential of Mamba (ICLR 2025)☆31Updated 7 months ago
- ☆61Updated 4 months ago
- ☆33Updated 9 months ago
- [ICLR 2025] Official Pytorch Implementation of "Mix-LN: Unleashing the Power of Deeper Layers by Combining Pre-LN and Post-LN" by Pengxia…☆27Updated 3 months ago
- The official repository for SkyLadder: Better and Faster Pretraining via Context Window Scheduling☆40Updated last month
- LongLLaDA: Unlocking Long Context Capabilities in Diffusion LLMs☆44Updated 3 months ago
- Stick-breaking attention☆61Updated 4 months ago
- ☆103Updated 2 months ago
- [ICLR 2025] When Attention Sink Emerges in Language Models: An Empirical View (Spotlight)☆135Updated 4 months ago
- Optimizing Anytime Reasoning via Budget Relative Policy Optimization☆47Updated 4 months ago
- ☆35Updated last year
- ☆88Updated last year
- Code for "Everybody Prune Now: Structured Pruning of LLMs with only Forward Passes"☆28Updated last year
- ☆19Updated 10 months ago
- Code for paper "Diffusion Language Models Can Perform Many Tasks with Scaling and Instruction-Finetuning"☆83Updated last year
- ☆19Updated 10 months ago
- ☆45Updated last month
- [ICLR 2023] "Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers" by Tianlong Chen*, Zhenyu Zhang*, Ajay Jaiswal…☆55Updated 2 years ago