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.
☆96Updated last year
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☆502Updated 7 months ago
- [ICCV2025] Introduce Mamba2 to Vision.☆162Updated 3 months ago
- Code Implementation of EfficientVMamba☆230Updated last year
- Code for paper LocalMamba: Visual State Space Model with Windowed Selective Scan☆266Updated last year
- The official repo for [TPAMI'23] "Vision Transformer with Quadrangle Attention"☆221Updated last week
- Official repository of MLLA (NeurIPS 2024)☆353Updated 2 months ago
- [ICML 2024] Official PyTorch implementation of "SLAB: Efficient Transformers with Simplified Linear Attention and Progressive Re-paramete…☆107Updated last year
- Awesome list of papers that extend Mamba to various applications.☆138Updated 3 months ago
- Official repository of Slide-Transformer (CVPR2023)☆170Updated last year
- Ofiicial Implementation for Mamba-ND: Selective State Space Modeling for Multi-Dimensional Data☆64Updated last year
- GroupMamba: Parameter-Efficient and Accurate Group Visual State Space Model [CVPR -2025]☆121Updated 6 months ago
- Official repository of Polarity-aware Linear Attention for Vision Transformers (ICLR 2025)☆71Updated 4 months ago
- Official PyTorch Implementation of "The Hidden Attention of Mamba Models"☆227Updated last year
- [ICLR2025] This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision☆85Updated 4 months ago
- A Triton Kernel for incorporating Bi-Directionality in Mamba2☆75Updated 9 months ago
- ☆47Updated last year
- Mamba in Vision: A Comprehensive Survey of Techniques and Applications☆125Updated 11 months ago
- Official repository of InLine attention (NeurIPS 2024)☆55Updated 9 months ago
- Minimal Mamba-2 implementation in PyTorch☆221Updated 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…☆223Updated last year
- [ICLR2025] Spatial-Mamba: Effective Visual State Space Models via Structure-Aware State Fusion☆171Updated 7 months ago
- Causal depthwise conv1d in CUDA, with a PyTorch interface☆605Updated last month
- [NeurIPS2024 Spotlight] The official implementation of MambaTree: Tree Topology is All You Need in State Space Model☆102Updated last year
- ☆73Updated 7 months ago
- PyTorch reimplementation of FlexiViT: One Model for All Patch Sizes☆62Updated last year
- Simba☆213Updated last year
- [BMVC 2024] PlainMamba: Improving Non-hierarchical Mamba in Visual Recognition☆83Updated 6 months ago
- (CVPR2024)RMT: Retentive Networks Meet Vision Transformer☆366Updated last year
- ✨✨Latest Papers on Vision Mamba and Related Areas☆368Updated 5 months ago
- Implementation of Vision Mamba from the paper: "Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Mod…☆467Updated 2 weeks ago