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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A simple and efficient Mamba implementation in pure PyTorch and MLX.☆1,442Jan 26, 2026Updated last month
- VMamba: Visual State Space Models，code is based on mamba☆3,085Mar 7, 2025Updated last year
- Code for paper LocalMamba: Visual State Space Model with Windowed Selective Scan☆277May 6, 2024Updated last year
- ☆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
- Export VMamba to onnx. VMamba: Visual State Space Models，code is based on VMamba: https://github.com/MzeroMiko/VMamba☆22May 13, 2025Updated 10 months ago
- Implementation of MoE Mamba from the paper: "MoE-Mamba: Efficient Selective State Space Models with Mixture of Experts" in Pytorch and Ze…☆124Updated this week
- [CVPR 2025] DynRefer: Delving into Region-level Multimodal Tasks via Dynamic Resolution☆59Mar 4, 2025Updated last year
- ☆11Aug 17, 2023Updated 2 years ago
- Simple, minimal implementation of the Mamba SSM in one file of PyTorch.☆2,931Mar 8, 2024Updated 2 years ago
- Implementation of paper "CC-Diff: Enhancing Contextual Coherence in Remote Sensing Image Synthesis"☆28Dec 19, 2025Updated 3 months ago
- Reading list for research topics in state-space models☆354Jun 11, 2025Updated 9 months ago
- Awesome Papers related to Mamba.☆1,393Oct 17, 2024Updated last year
- ☆26Oct 15, 2024Updated last year
- The official implementation of BiViT: Extremely Compressed Binary Vision Transformers☆16Jun 18, 2023Updated 2 years ago
- [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
- [ICLR2025] This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision☆90May 30, 2025Updated 9 months ago
- Implementation of "VL-Mamba: Exploring State Space Models for Multimodal Learning"☆86Mar 21, 2024Updated 2 years ago
- [NeurIPS2024 Spotlight] The official implementation of MambaTree: Tree Topology is All You Need in State Space Model☆107Jun 13, 2024Updated last year
- [NeurIPS 2024] State Space Models on Temporal Graphs: A First-Principles Study☆16Dec 31, 2024Updated last year
- Official source code of "Stable and Consistent Prediction of 3D Characteristic Orientation vis Invariant Residual Learning", ICML 2023☆14Jul 22, 2023Updated 2 years ago
- FlashRNN - Fast RNN Kernels with I/O Awareness☆177Oct 20, 2025Updated 5 months 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
- ☆58Jul 9, 2024Updated last year
- Official PyTorch implementation of the Winner Award solution of MIPI 2024 Demosaic for HybridEVS Camera Challenge (CVPR Workshop 2024).☆16Jan 28, 2025Updated last year
- CVPR2024, Semantic-aware SAM for Point-Prompted Instance Segmentation☆37Jan 20, 2025Updated last year
- ☆13Mar 2, 2023Updated 3 years ago
- [ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model☆3,823Feb 13, 2025Updated last year
- ☆35Nov 22, 2024Updated last year
- AIRS-2025赛道二：「星际矿脉」火星矿物高光谱分类挑战赛☆12May 7, 2025Updated 10 months ago
- Official PyTorch Implementation of Efficient and Versatile Robust Fine-Tuning of Zero-shot Models, ECCV 2024☆17Oct 3, 2024Updated last year
- ☆107Mar 9, 2024Updated 2 years ago
- Implementation of a modular, high-performance, and simplistic mamba for high-speed applications☆40Nov 11, 2024Updated last year
- Multi-Resolution Neural Networks☆19Mar 5, 2026Updated 2 weeks ago
- Swin-UMamba: Mamba-based UNet with ImageNet-based pretraining☆376Mar 19, 2024Updated 2 years ago
- A paper list of some recent Mamba-based CV works.☆457Mar 3, 2026Updated 2 weeks ago