assafbk / DeciMamba
DeciMamba: Exploring the Length Extrapolation Potential of Mamba (ICLR 2025)
☆26Updated 3 weeks ago
Alternatives and similar repositories for DeciMamba:
Users that are interested in DeciMamba are comparing it to the libraries listed below
- HGRN2: Gated Linear RNNs with State Expansion☆54Updated 8 months ago
- [ICML 2024] When Linear Attention Meets Autoregressive Decoding: Towards More Effective and Efficient Linearized Large Language Models☆30Updated 10 months ago
- ☆39Updated last month
- ☆53Updated 9 months ago
- Stick-breaking attention☆52Updated last month
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆64Updated last year
- Official code for the paper "Attention as a Hypernetwork"☆30Updated 10 months ago
- ☆31Updated last year
- [ICML 2024 Oral] This project is the official implementation of our Accurate LoRA-Finetuning Quantization of LLMs via Information Retenti…☆65Updated last year
- ☆15Updated last year
- ☆22Updated last month
- Official implementation of the ICML 2024 paper RoSA (Robust Adaptation)☆41Updated last year
- [ICLR 2025] Official PyTorch implementation of "Forgetting Transformer: Softmax Attention with a Forget Gate"☆97Updated 3 weeks ago
- ☆23Updated 7 months ago
- ☆78Updated 8 months ago
- Triton implement of bi-directional (non-causal) linear attention☆46Updated 3 months ago
- The this is the official implementation of "DAPE: Data-Adaptive Positional Encoding for Length Extrapolation"☆37Updated 6 months ago
- A Triton Kernel for incorporating Bi-Directionality in Mamba2☆65Updated 4 months ago
- MambaFormer in-context learning experiments and implementation for https://arxiv.org/abs/2402.04248☆52Updated 10 months ago
- A repository for DenseSSMs☆87Updated last year
- [ICLR 2023] "Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers" by Tianlong Chen*, Zhenyu Zhang*, Ajay Jaiswal…☆51Updated 2 years ago
- ☆69Updated 2 months ago
- An unofficial implementation of "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"☆35Updated 10 months ago
- Official implementation of Phi-Mamba. A MOHAWK-distilled model (Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Mode…☆106Updated 7 months ago
- Unofficial Implementation of Selective Attention Transformer☆16Updated 6 months ago
- [ICLR 2025] Official Code Release for Explaining Modern Gated-Linear RNNs via a Unified Implicit Attention Formulation☆42Updated 2 months ago
- ☆80Updated 3 weeks ago
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆58Updated last month
- "Found in the Middle: How Language Models Use Long Contexts Better via Plug-and-Play Positional Encoding" Zhenyu Zhang, Runjin Chen, Shiw…☆29Updated 11 months ago
- Code for "Everybody Prune Now: Structured Pruning of LLMs with only Forward Passes"☆28Updated last year