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.
☆93Updated 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:
- Code Implementation of EfficientVMamba☆216Updated last year
- [ICLR 2025 Spotlight] Vision-RWKV: Efficient and Scalable Visual Perception with RWKV-Like Architectures☆478Updated 5 months ago
- [ICCV2025] Introduce Mamba2 to Vision.☆141Updated 3 weeks ago
- Official PyTorch Implementation of "The Hidden Attention of Mamba Models"☆224Updated last year
- Code for paper LocalMamba: Visual State Space Model with Windowed Selective Scan☆258Updated last year
- Official repository of MLLA (NeurIPS 2024)☆335Updated last week
- [ICML 2024] Official PyTorch implementation of "SLAB: Efficient Transformers with Simplified Linear Attention and Progressive Re-paramete…☆106Updated 10 months ago
- Official repository of Slide-Transformer (CVPR2023)☆171Updated 10 months ago
- Awesome list of papers that extend Mamba to various applications.☆134Updated last month
- GroupMamba: Parameter-Efficient and Accurate Group Visual State Space Model [CVPR -2025]☆105Updated 3 months ago
- [ICLR2025] This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision☆83Updated last month
- The official repo for [TPAMI'23] "Vision Transformer with Quadrangle Attention"☆217Updated last year
- Ofiicial Implementation for Mamba-ND: Selective State Space Modeling for Multi-Dimensional Data☆61Updated last year
- A Triton Kernel for incorporating Bi-Directionality in Mamba2☆71Updated 7 months ago
- ☆69Updated 4 months ago
- Minimal Mamba-2 implementation in PyTorch☆207Updated last year
- [NeurIPS2024 Spotlight] The official implementation of MambaTree: Tree Topology is All You Need in State Space Model☆96Updated last year
- Official repository of Polarity-aware Linear Attention for Vision Transformers (ICLR 2025)☆67Updated 2 months ago
- Causal depthwise conv1d in CUDA, with a PyTorch interface☆516Updated this week
- Open source implementation of "Vision Transformers Need Registers"☆184Updated 3 months ago
- This repository contains the pytorch code for our work IEEE ISBI 2024 paper "ConvLoRA and AdaBN Based Domain Adaptation via Self-Training…☆80Updated 9 months ago
- Mamba in Vision: A Comprehensive Survey of Techniques and Applications☆121Updated 9 months ago
- Computation-Efficient Era: A Comprehensive Survey of State Space Models in Medical Image Analysis☆241Updated 5 months ago
- [BMVC 2024] PlainMamba: Improving Non-hierarchical Mamba in Visual Recognition☆79Updated 3 months ago
- PyTorch reimplementation of FlexiViT: One Model for All Patch Sizes☆62Updated last year
- Official Implementation of the CrossMAE paper: Rethinking Patch Dependence for Masked Autoencoders☆115Updated 3 months ago
- [CVPR'24] Multimodal Pathway: Improve Transformers with Irrelevant Data from Other Modalities☆99Updated last year
- ☆69Updated 5 months ago
- Official repository of InLine attention (NeurIPS 2024)☆49Updated 6 months ago
- [ICLR2025] Spatial-Mamba: Effective Visual State Space Models via Structure-Aware State Fusion☆158Updated 4 months ago