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 6 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,450Jan 26, 2026Updated 2 months ago
- VMamba: Visual State Space Models，code is based on mamba☆3,120Mar 7, 2025Updated last year
- Code for paper LocalMamba: Visual State Space Model with Windowed Selective Scan☆278May 6, 2024Updated last year
- Modified Mamba code to run on CPU☆30Jan 14, 2024Updated 2 years ago
- ☆15Oct 14, 2025Updated 6 months ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Official PyTorch Implementation of "The Hidden Attention of Mamba Models"☆232Oct 16, 2025Updated 5 months ago
- [CVPR 2024] Learning SO(3)-Invariant Semantic Correspondence via Local Shape Transform☆14Updated this week
- Export VMamba to onnx. VMamba: Visual State Space Models，code is based on VMamba: https://github.com/MzeroMiko/VMamba☆22May 13, 2025Updated 11 months ago
- Implementation of MoE Mamba from the paper: "MoE-Mamba: Efficient Selective State Space Models with Mixture of Experts" in Pytorch and Ze…☆126Mar 22, 2026Updated 3 weeks ago
- A Triton Kernel for incorporating Bi-Directionality in Mamba2☆82Dec 18, 2024Updated last year
- Simple, minimal implementation of the Mamba SSM in one file of PyTorch.☆2,944Mar 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☆357Jun 11, 2025Updated 10 months ago
- Awesome Papers related to Mamba.☆1,395Oct 17, 2024Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- ☆26Oct 15, 2024Updated last year
- [Mamba-Survey-2024] Paper list for State-Space-Model/Mamba and it's Applications☆749Jun 28, 2025Updated 9 months ago
- Some preliminary explorations of Mamba's context scaling.☆219Feb 8, 2024Updated 2 years ago
- [ICLR2025] This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision☆90May 30, 2025Updated 10 months ago
- More dimensions = More fun☆26Jul 27, 2024Updated last year
- Implementation of "VL-Mamba: Exploring State Space Models for Multimodal Learning"☆86Mar 21, 2024Updated 2 years ago
- [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
- FlashRNN - Fast RNN Kernels with I/O Awareness☆179Oct 20, 2025Updated 5 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Implementation of mamba with rust☆92Mar 9, 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
- Notes on the Mamba and the S4 model (Mamba: Linear-Time Sequence Modeling with Selective State Spaces)☆181Jan 7, 2024Updated 2 years ago
- ☆58Jul 9, 2024Updated last year
- Official repository for the paper "Instance-Wise Holistic Order Prediction in Natural Scenes".☆27Jan 11, 2024Updated 2 years ago
- AIRS-2025赛道二：「星际矿脉」火星矿物高光谱分类挑战赛☆12May 7, 2025Updated 11 months ago
- ☆35Nov 22, 2024Updated last year
- [ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model☆3,839Feb 13, 2025Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- ☆107Mar 9, 2024Updated 2 years ago
- ☆51Jan 28, 2024Updated 2 years ago
- Implementation of a modular, high-performance, and simplistic mamba for high-speed applications☆40Nov 11, 2024Updated last year
- Swin-UMamba: Mamba-based UNet with ImageNet-based pretraining☆379Mar 19, 2024Updated 2 years ago
- A paper list of some recent Mamba-based CV works.☆467Mar 3, 2026Updated last month
- ☆11Feb 9, 2026Updated 2 months ago
- Inference of Mamba, Mamba2 and Mamba3 models in pure C☆199Mar 18, 2026Updated 3 weeks ago