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.
☆91Updated 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:
- A Triton Kernel for incorporating Bi-Directionality in Mamba2☆70Updated 6 months ago
- Awesome list of papers that extend Mamba to various applications.☆133Updated 2 weeks ago
- Code Implementation of EfficientVMamba☆213Updated last year
- Open source implementation of "Vision Transformers Need Registers"☆182Updated 2 months ago
- Code for paper LocalMamba: Visual State Space Model with Windowed Selective Scan☆254Updated last year
- [ICLR2025] This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision☆80Updated 3 weeks ago
- ☆69Updated 4 months ago
- [ICLR 2025 Spotlight] Vision-RWKV: Efficient and Scalable Visual Perception with RWKV-Like Architectures☆472Updated 4 months ago
- Introduce Mamba2 to Vision.☆132Updated 10 months ago
- [NeurIPS2024 Spotlight] The official implementation of MambaTree: Tree Topology is All You Need in State Space Model☆93Updated last year
- Official PyTorch Implementation of "The Hidden Attention of Mamba Models"☆223Updated last year
- Mamba in Vision: A Comprehensive Survey of Techniques and Applications☆121Updated 8 months ago
- GroupMamba: Parameter-Efficient and Accurate Group Visual State Space Model [CVPR -2025]☆98Updated 3 months ago
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆138Updated 4 months ago
- Ofiicial Implementation for Mamba-ND: Selective State Space Modeling for Multi-Dimensional Data☆60Updated 11 months ago
- Official repository of MLLA (NeurIPS 2024)☆330Updated 7 months ago
- Official repository of InLine attention (NeurIPS 2024)☆49Updated 6 months ago
- Official repository of Polarity-aware Linear Attention for Vision Transformers (ICLR 2025)☆64Updated last month
- A PyTorch implementation of the paper "ZigMa: A DiT-Style Mamba-based Diffusion Model" (ECCV 2024)☆318Updated 3 months ago
- A repository for DenseSSMs☆87Updated 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…☆221Updated 10 months ago
- [ICLR2025] Spatial-Mamba: Effective Visual State Space Models via Structure-Aware State Fusion☆150Updated 3 months ago
- The official repo for [TPAMI'23] "Vision Transformer with Quadrangle Attention"☆217Updated last year
- Causal depthwise conv1d in CUDA, with a PyTorch interface☆498Updated this week
- A library for calculating the FLOPs in the forward() process based on torch.fx☆115Updated 2 months ago
- ☆63Updated 4 months ago
- Minimal Mamba-2 implementation in PyTorch☆202Updated last year
- A curated list of papers on the applications of RWKV in computer vision.☆191Updated last week
- The official implementation of DiM: Diffusion Mamba for Efficient High-Resolution Image Synthesis☆197Updated 11 months ago
- [BMVC 2024] PlainMamba: Improving Non-hierarchical Mamba in Visual Recognition☆78Updated 2 months ago