tim-lawson / skip-middleLinks
Learning to Skip the Middle Layers of Transformers
☆17Updated 5 months ago
Alternatives and similar repositories for skip-middle
Users that are interested in skip-middle are comparing it to the libraries listed below
Sorting:
- Unofficial Implementation of Selective Attention Transformer☆20Updated last year
- ☆40Updated 4 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…☆29Updated 6 months ago
- [ICLR 2025] Drop-Upcycling: Training Sparse Mixture of Experts with Partial Re-initialization☆23Updated 3 months ago
- AdaSplash: Adaptive Sparse Flash Attention (aka Flash Entmax Attention)☆32Updated 4 months ago
- [ICML 2024] When Linear Attention Meets Autoregressive Decoding: Towards More Effective and Efficient Linearized Large Language Models☆35Updated last year
- Official implementation of the paper: "A deeper look at depth pruning of LLMs"☆15Updated last year
- [ICML 2025] Code for "R2-T2: Re-Routing in Test-Time for Multimodal Mixture-of-Experts"☆17Updated 10 months ago
- [NeurIPS '25] Multi-Token Prediction Needs Registers☆26Updated last month
- ☆36Updated 10 months ago
- HGRN2: Gated Linear RNNs with State Expansion☆56Updated last year
- [NAACL'25 🏆 SAC Award] Official code for "Advancing MoE Efficiency: A Collaboration-Constrained Routing (C2R) Strategy for Better Expert…☆14Updated 11 months ago
- [AAAI26] LongLLaDA: Unlocking Long Context Capabilities in Diffusion LLMs☆51Updated last month
- ☆27Updated 2 months ago
- The official repo of continuous speculative decoding☆31Updated 10 months ago
- Xmixers: A collection of SOTA efficient token/channel mixers☆28Updated 4 months ago
- M1: Towards Scalable Test-Time Compute with Mamba Reasoning Models☆47Updated 6 months ago
- Official Implementation of FastKV: Decoupling of Context Reduction and KV Cache Compression for Prefill-Decoding Acceleration☆28Updated 2 months ago
- [ICLR 2025 & COLM 2025] Official PyTorch implementation of the Forgetting Transformer and Adaptive Computation Pruning☆137Updated last month
- KVTuner: Sensitivity-Aware Layer-wise Mixed Precision KV Cache Quantization for Efficient and Nearly Lossless LLM Inference☆26Updated 8 months ago
- Triton implement of bi-directional (non-causal) linear attention☆63Updated 11 months ago
- [ICLR 2025] Linear Combination of Saved Checkpoints Makes Consistency and Diffusion Models Better☆16Updated 11 months ago
- ☆73Updated 7 months ago
- [NeurIPS 2025] Official implementation of "Reasoning Path Compression: Compressing Generation Trajectories for Efficient LLM Reasoning"☆29Updated 3 months ago
- [ICML 2025] LaCache: Ladder-Shaped KV Caching for Efficient Long-Context Modeling of Large Language Models☆17Updated 2 months ago
- ☆17Updated 7 months ago
- ☆15Updated last year
- codes and plots for "Active-Dormant Attention Heads: Mechanistically Demystifying Extreme-Token Phenomena in LLMs"☆10Updated last year
- Are gradient information useful for pruning of LLMs?☆47Updated 5 months ago
- ☆19Updated last year