tim-lawson / skip-middleLinks
Learning to Skip the Middle Layers of Transformers
☆17Updated 6 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
- Official implementation of the paper: "A deeper look at depth pruning of LLMs"☆15Updated last year
- ☆36Updated 11 months ago
- ☆42Updated 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
- [NeurIPS '25] Multi-Token Prediction Needs Registers☆26Updated last month
- [ICML 2025] Code for "R2-T2: Re-Routing in Test-Time for Multimodal Mixture-of-Experts"☆19Updated 11 months ago
- AdaSplash: Adaptive Sparse Flash Attention (aka Flash Entmax Attention)☆32Updated 4 months ago
- User-friendly implementation of the Mixture-of-Sparse-Attention (MoSA). MoSA selects distinct tokens for each head with expert choice rou…☆28Updated 9 months ago
- ☆15Updated last year
- [ICLR 2025] Drop-Upcycling: Training Sparse Mixture of Experts with Partial Re-initialization☆24Updated 4 months ago
- [ICML 2024] When Linear Attention Meets Autoregressive Decoding: Towards More Effective and Efficient Linearized Large Language Models☆35Updated last year
- ☆75Updated 7 months ago
- Official PyTorch Implementation for Vision-Language Models Create Cross-Modal Task Representations, ICML 2025☆32Updated 9 months ago
- The official repository for SkyLadder: Better and Faster Pretraining via Context Window Scheduling☆42Updated last month
- Triton implement of bi-directional (non-causal) linear attention☆65Updated last week
- [ICML 2025] LaCache: Ladder-Shaped KV Caching for Efficient Long-Context Modeling of Large Language Models☆17Updated 3 months ago
- Control LLM☆22Updated 10 months ago
- ☆19Updated 10 months ago
- ☆19Updated last year
- The official repo of continuous speculative decoding☆31Updated 10 months ago
- Xmixers: A collection of SOTA efficient token/channel mixers☆28Updated 5 months ago
- HGRN2: Gated Linear RNNs with State Expansion☆56Updated last year
- Resa: Transparent Reasoning Models via SAEs☆47Updated 4 months ago
- Official PyTorch implementation and models for paper "Diffusion Beats Autoregressive in Data-Constrained Settings". We find diffusion mod…☆120Updated last month
- [AAAI26] LongLLaDA: Unlocking Long Context Capabilities in Diffusion LLMs☆52Updated 2 months ago
- [ICML 2024] Official Repository for the paper "Transformers Get Stable: An End-to-End Signal Propagation Theory for Language Models"☆10Updated last year
- Mixture-of-Basis-Experts for Compressing MoE-based LLMs☆27Updated last month
- Official code for the paper "Attention as a Hypernetwork"☆47Updated last year
- Official Implementation of FastKV: Decoupling of Context Reduction and KV Cache Compression for Prefill-Decoding Acceleration☆29Updated 2 months ago