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.
☆103Oct 14, 2025Updated 4 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,046Mar 7, 2025Updated 11 months ago
- Code for paper LocalMamba: Visual State Space Model with Windowed Selective Scan☆274May 6, 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…☆120Jan 31, 2026Updated last month
- Official PyTorch Implementation of "The Hidden Attention of Mamba Models"☆231Oct 16, 2025Updated 4 months ago
- Implementation of a modular, high-performance, and simplistic mamba for high-speed applications☆40Nov 11, 2024Updated last year
- ☆35Nov 22, 2024Updated last year
- [Mamba-Survey-2024] Paper list for State-Space-Model/Mamba and it's Applications☆745Jun 28, 2025Updated 8 months ago
- Reading list for research topics in state-space models☆347Jun 11, 2025Updated 8 months ago
- Some preliminary explorations of Mamba's context scaling.☆218Feb 8, 2024Updated 2 years ago
- A Triton Kernel for incorporating Bi-Directionality in Mamba2☆78Dec 18, 2024Updated last year
- Awesome Papers related to Mamba.☆1,390Oct 17, 2024Updated last year
- [CVPR 2024] Learning SO(3)-Invariant Semantic Correspondence via Local Shape Transform☆11Jun 15, 2024Updated last year
- Swin-UMamba: Mamba-based UNet with ImageNet-based pretraining☆371Mar 19, 2024Updated last year
- ☆26Oct 15, 2024Updated last year
- Notes on the Mamba and the S4 model (Mamba: Linear-Time Sequence Modeling with Selective State Spaces)☆179Jan 7, 2024Updated 2 years ago
- Awesome list of papers that extend Mamba to various applications.☆139Jun 10, 2025Updated 8 months ago
- ☆51Jan 28, 2024Updated 2 years ago
- ☆106Mar 9, 2024Updated last year
- ☆11Aug 17, 2023Updated 2 years ago
- ☆12Mar 25, 2024Updated last year
- A simpler Pytorch + Zeta Implementation of the paper: "SiMBA: Simplified Mamba-based Architecture for Vision and Multivariate Time series…☆28Nov 11, 2024Updated last year
- Simple, minimal implementation of the Mamba SSM in one file of PyTorch.☆2,921Mar 8, 2024Updated last year
- Collect papers about Mamba (a selective state space model).☆14Aug 6, 2024Updated last year
- Here we will test various linear attention designs.☆62Apr 25, 2024Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆93Jan 25, 2024Updated 2 years ago
- [ICLR2025] This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision☆90May 30, 2025Updated 9 months ago
- Implementation of mamba with rust☆92Mar 9, 2024Updated last year
- Official code for "Audio-Guided Attention Network for Weakly Supervised Violence Detection" (ICCECE2022).☆13Mar 25, 2022Updated 3 years ago
- This repository contains the supplementary material for the paper titled: "TRANSFORMER COMPRESSED SENSING VIA GLOBAL IMAGE TOKENS".☆13Dec 20, 2022Updated 3 years ago
- ✨✨Latest Papers on Vision Mamba and Related Areas☆382Apr 17, 2025Updated 10 months ago
- ☆23Jun 13, 2023Updated 2 years ago
- Official source code of "Stable and Consistent Prediction of 3D Characteristic Orientation vis Invariant Residual Learning", ICML 2023☆14Jul 22, 2023Updated 2 years ago
- A program for panoramic image stitching based on opencv☆13Mar 3, 2021Updated 4 years ago
- [ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model☆3,799Feb 13, 2025Updated last year
- [NeurIPS2024 Spotlight] The official implementation of MambaTree: Tree Topology is All You Need in State Space Model☆105Jun 13, 2024Updated last year
- Official PyTorch implementation of the Winner Award solution of MIPI 2024 Demosaic for HybridEVS Camera Challenge (CVPR Workshop 2024).☆15Jan 28, 2025Updated last year
- Pytorch implementation of "DAMA - Multiplexed Immunofluorescence Brain Image Analysis Using Self-Supervised Dual-Loss Adaptive Masked Aut…☆17Oct 20, 2023Updated 2 years ago
- Integrating Mamba/SSMs with Transformer for Enhanced Long Context and High-Quality Sequence Modeling☆215Jan 30, 2026Updated last month
- ☆69Oct 22, 2024Updated last year