MzeroMiko / mamba-miniLinks
An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivation. It is probably the code which is the most close to selective_scan_cuda in mamba.
☆100Updated 2 months ago
Alternatives and similar repositories for mamba-mini
Users that are interested in mamba-mini are comparing it to the libraries listed below
Sorting:
- [ICLR 2025 Spotlight] Vision-RWKV: Efficient and Scalable Visual Perception with RWKV-Like Architectures☆530Updated 10 months ago
- Code Implementation of EfficientVMamba☆238Updated last year
- Official PyTorch Implementation of "The Hidden Attention of Mamba Models"☆231Updated 2 months ago
- Awesome list of papers that extend Mamba to various applications.☆139Updated 6 months ago
- Code for paper LocalMamba: Visual State Space Model with Windowed Selective Scan☆273Updated last year
- [ICCV2025] Introduce Mamba2 to Vision.☆177Updated last month
- The official repo for [TPAMI'23] "Vision Transformer with Quadrangle Attention"☆227Updated 2 months ago
- A Triton Kernel for incorporating Bi-Directionality in Mamba2☆75Updated last year
- Official repository of MLLA (NeurIPS 2024)☆367Updated 5 months ago
- GroupMamba: Parameter-Efficient and Accurate Group Visual State Space Model [CVPR -2025]☆128Updated 9 months ago
- [ICLR2025] This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision☆88Updated 6 months ago
- Official repository of Slide-Transformer (CVPR2023)☆174Updated last year
- [ICML 2024] Official PyTorch implementation of "SLAB: Efficient Transformers with Simplified Linear Attention and Progressive Re-paramete…☆109Updated last year
- Official repository of Polarity-aware Linear Attention for Vision Transformers (ICLR 2025)☆80Updated last month
- ☆78Updated 9 months ago
- Mamba in Vision: A Comprehensive Survey of Techniques and Applications☆130Updated last year
- Minimal Mamba-2 implementation in PyTorch☆237Updated last year
- Simba☆215Updated last year
- Ofiicial Implementation for Mamba-ND: Selective State Space Modeling for Multi-Dimensional Data☆65Updated last year
- 1.5−3.0× lossless training or pre-training speedup. An off-the-shelf, easy-to-implement algorithm for the efficient training of foundatio…☆226Updated last year
- Implementation of Vision Mamba from the paper: "Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Mod…☆480Updated 3 weeks ago
- ☆48Updated last year
- [NeurIPS2024 Spotlight] The official implementation of MambaTree: Tree Topology is All You Need in State Space Model☆104Updated last year
- ☆77Updated 10 months ago
- Official repository of InLine attention (NeurIPS 2024)☆57Updated last year
- [CVPR 2024] Multimodal Pathway: Improve Transformers with Irrelevant Data from Other Modalities☆101Updated last year
- Causal depthwise conv1d in CUDA, with a PyTorch interface☆675Updated 2 months ago
- Official Implementation of the CrossMAE paper: Rethinking Patch Dependence for Masked Autoencoders☆128Updated 8 months ago
- A repository for DenseSSMs☆89Updated last year
- [ICLR2025] Spatial-Mamba: Effective Visual State Space Models via Structure-Aware State Fusion☆186Updated 9 months ago