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.
☆101Updated 3 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:
- Code Implementation of EfficientVMamba☆243Updated last year
- [ICCV2025] Introduce Mamba2 to Vision.☆182Updated 3 months ago
- [ICLR2025] This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision☆90Updated 8 months ago
- Official PyTorch Implementation of "The Hidden Attention of Mamba Models"☆231Updated 3 months ago
- [ICLR 2025 Spotlight] Vision-RWKV: Efficient and Scalable Visual Perception with RWKV-Like Architectures☆540Updated 11 months ago
- Code for paper LocalMamba: Visual State Space Model with Windowed Selective Scan☆273Updated last year
- Awesome list of papers that extend Mamba to various applications.☆138Updated 7 months ago
- GroupMamba: Parameter-Efficient and Accurate Group Visual State Space Model [CVPR -2025]☆131Updated 10 months ago
- The official repo for [TPAMI'23] "Vision Transformer with Quadrangle Attention"☆234Updated 4 months ago
- Official repository of MLLA (NeurIPS 2024)☆370Updated 6 months ago
- Official repository of Slide-Transformer (CVPR2023)☆175Updated last year
- A Triton Kernel for incorporating Bi-Directionality in Mamba2☆78Updated last year
- [ICML 2024] Official PyTorch implementation of "SLAB: Efficient Transformers with Simplified Linear Attention and Progressive Re-paramete…☆108Updated last year
- Official repository of Polarity-aware Linear Attention for Vision Transformers (ICLR 2025)☆83Updated 3 months ago
- Minimal Mamba-2 implementation in PyTorch☆243Updated last year
- ☆78Updated 11 months ago
- [NeurIPS2024 Spotlight] The official implementation of MambaTree: Tree Topology is All You Need in State Space Model☆105Updated last year
- Mamba in Vision: A Comprehensive Survey of Techniques and Applications☆133Updated last year
- Ofiicial Implementation for Mamba-ND: Selective State Space Modeling for Multi-Dimensional Data☆66Updated last year
- Simba☆216Updated last year
- Open source implementation of "Vision Transformers Need Registers"☆209Updated 2 weeks ago
- 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
- This repository contains the pytorch code for our work IEEE ISBI 2024 paper "ConvLoRA and AdaBN Based Domain Adaptation via Self-Training…☆94Updated last year
- ☆69Updated last year
- PyTorch implementation of the Differential-Transformer architecture for sequence modeling, specifically tailored as a decoder-only model …☆85Updated last year
- [ICLR2025] Spatial-Mamba: Effective Visual State Space Models via Structure-Aware State Fusion☆194Updated 11 months ago
- PyTorch reimplementation of FlexiViT: One Model for All Patch Sizes☆65Updated last year
- ☆152Updated last year
- Official Implementation of the CrossMAE paper: Rethinking Patch Dependence for Masked Autoencoders☆131Updated 9 months ago
- Official repository of InLine attention (NeurIPS 2024)☆58Updated last year