Hprairie / Bi-Mamba2
A Triton Kernel for incorporating Bi-Directionality in Mamba2
☆47Updated 2 months ago
Related projects ⓘ
Alternatives and complementary repositories for Bi-Mamba2
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆102Updated 3 months ago
- Official PyTorch Implementation of "The Hidden Attention of Mamba Models"☆199Updated 5 months ago
- Implementation of Griffin from the paper: "Griffin: Mixing Gated Linear Recurrences with Local Attention for Efficient Language Models"☆49Updated this week
- A repository for DenseSSMs☆88Updated 6 months ago
- Official implementation of Phi-Mamba. A MOHAWK-distilled model (Transformers to SSMs: Distilling Quadratic Knowledge to Subquadratic Mode…☆76Updated last month
- Minimal Mamba-2 implementation in PyTorch☆127Updated 4 months ago
- An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivatio…☆71Updated 8 months ago
- The this is the official implementation of "DAPE: Data-Adaptive Positional Encoding for Length Extrapolation"☆30Updated 3 weeks ago
- Awesome list of papers that extend Mamba to various applications.☆127Updated last month
- ☆41Updated 7 months ago
- Ofiicial Implementation for Mamba-ND: Selective State Space Modeling for Multi-Dimensional Data☆49Updated 4 months ago
- State Space Models☆61Updated 6 months ago
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆61Updated 6 months ago
- My implementation of the original transformer model (Vaswani et al.). I've additionally included the playground.py file for visualizing o…☆40Updated 11 months ago
- Pytorch Implementation of the paper: "Learning to (Learn at Test Time): RNNs with Expressive Hidden States"☆23Updated this week
- ☆37Updated last month
- Official Implementation Of The Paper: `DeciMamba: Exploring the Length Extrapolation Potential of Mamba'☆20Updated 3 months ago
- Reading list for research topics in state-space models☆237Updated last week
- HGRN2: Gated Linear RNNs with State Expansion☆48Updated 2 months ago
- Implementation of Agent Attention in Pytorch☆86Updated 3 months ago
- Implementation of the paper: "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"☆67Updated this week
- Simba☆182Updated 7 months ago
- Stick-breaking attention☆32Updated last week
- Here we will test various linear attention designs.☆56Updated 6 months ago
- More dimensions = More fun☆21Updated 3 months ago
- Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)☆63Updated 6 months ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆94Updated 2 weeks ago
- Integrating Mamba/SSMs with Transformer for Enhanced Long Context and High-Quality Sequence Modeling☆167Updated this week
- PyTorch Implementation of Jamba: "Jamba: A Hybrid Transformer-Mamba Language Model"☆134Updated this week
- Inference Speed Benchmark for Learning to (Learn at Test Time): RNNs with Expressive Hidden States☆38Updated 3 months ago