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.
☆104Oct 14, 2025Updated 5 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:
- VMamba: Visual State Space Models，code is based on mamba☆3,079Mar 7, 2025Updated last year
- ☆15Oct 14, 2025Updated 5 months ago
- ☆31Sep 24, 2024Updated last year
- Official PyTorch Implementation of "The Hidden Attention of Mamba Models"☆231Oct 16, 2025Updated 5 months ago
- [CVPR 2024] Learning SO(3)-Invariant Semantic Correspondence via Local Shape Transform☆11Jun 15, 2024Updated last year
- Implementation of MoE Mamba from the paper: "MoE-Mamba: Efficient Selective State Space Models with Mixture of Experts" in Pytorch and Ze…☆122Jan 31, 2026Updated last month
- A Triton Kernel for incorporating Bi-Directionality in Mamba2☆78Dec 18, 2024Updated last year
- ☆11Aug 17, 2023Updated 2 years ago
- Awesome Papers related to Mamba.☆1,393Oct 17, 2024Updated last year
- ☆26Oct 15, 2024Updated last year
- [Mamba-Survey-2024] Paper list for State-Space-Model/Mamba and it's Applications☆748Jun 28, 2025Updated 8 months ago
- Some preliminary explorations of Mamba's context scaling.☆218Feb 8, 2024Updated 2 years ago
- More dimensions = More fun☆26Jul 27, 2024Updated last year
- [NeurIPS2024 Spotlight] The official implementation of MambaTree: Tree Topology is All You Need in State Space Model☆106Jun 13, 2024Updated last year
- [NeurIPS 2024] State Space Models on Temporal Graphs: A First-Principles Study☆16Dec 31, 2024Updated last year
- This repository contains the supplementary material for the paper titled: "TRANSFORMER COMPRESSED SENSING VIA GLOBAL IMAGE TOKENS".☆13Dec 20, 2022Updated 3 years ago
- Implementation of mamba with rust☆92Mar 9, 2024Updated 2 years ago
- Notes on the Mamba and the S4 model (Mamba: Linear-Time Sequence Modeling with Selective State Spaces)☆180Jan 7, 2024Updated 2 years ago
- ☆23Jun 17, 2025Updated 9 months ago
- Official PyTorch implementation of the Winner Award solution of MIPI 2024 Demosaic for HybridEVS Camera Challenge (CVPR Workshop 2024).☆16Jan 28, 2025Updated last year
- Official repository for the paper "Instance-Wise Holistic Order Prediction in Natural Scenes".☆27Jan 11, 2024Updated 2 years ago
- [ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model☆3,820Feb 13, 2025Updated last year
- ☆35Nov 22, 2024Updated last year
- ☆51Jan 28, 2024Updated 2 years ago
- ☆107Mar 9, 2024Updated 2 years ago
- Swin-UMamba: Mamba-based UNet with ImageNet-based pretraining☆375Mar 19, 2024Updated 2 years ago
- A paper list of some recent Mamba-based CV works.☆455Mar 3, 2026Updated 2 weeks ago
- ☆33Sep 6, 2024Updated last year
- Integrating Mamba/SSMs with Transformer for Enhanced Long Context and High-Quality Sequence Modeling☆217Jan 30, 2026Updated last month
- ✨✨Latest Papers on Vision Mamba and Related Areas☆383Apr 17, 2025Updated 11 months ago
- ☆11Feb 9, 2026Updated last month
- Inference of Mamba, Mamba2 and Mamba3 models in pure C☆199Updated this week
- Official Implementation of GENIUS: A Generative Framework for Universal Multimodal Search, CVPR 2025☆51Aug 8, 2025Updated 7 months ago
- [NeurIPS 2024] Official repository of MLLA☆373Jul 11, 2025Updated 8 months ago
- Here we will test various linear attention designs.☆62Apr 25, 2024Updated last year
- Code Implementation of EfficientVMamba☆244Apr 16, 2024Updated last year
- Official repository of ”Mamba-FSCIL: Dynamic Adaptation with Selective State Space Model for Few-Shot Class-Incremental Learning"☆39Aug 1, 2025Updated 7 months ago
- ☆12Mar 25, 2024Updated last year
- Mamba SSM architecture☆17,524Updated this week