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.
☆109Oct 14, 2025Updated 11 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,480May 3, 2026Updated 5 months ago
- VMamba: Visual State Space Models，code is based on mamba☆3,233Mar 7, 2025Updated last year
- Code for paper LocalMamba: Visual State Space Model with Windowed Selective Scan☆285May 6, 2024Updated 2 years ago
- [ICML 2026 Spotlight] Code for miXed Discrete Diffusion Language Model☆31Mar 16, 2026Updated 6 months ago
- Implementation of paper "CC-Diff: Enhancing Contextual Coherence in Remote Sensing Image Synthesis"☆29Dec 19, 2025Updated 9 months ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Modified Mamba code to run on CPU☆34Jan 14, 2024Updated 2 years ago
- Official PyTorch Implementation of "The Hidden Attention of Mamba Models"☆238Oct 16, 2025Updated 11 months ago
- Export VMamba to onnx. VMamba: Visual State Space Models，code is based on VMamba: https://github.com/MzeroMiko/VMamba☆25May 13, 2025Updated last year
- [CVPR 2024] Learning SO(3)-Invariant Semantic Correspondence via Local Shape Transform☆14Apr 9, 2026Updated 6 months ago
- Implementation of MoE Mamba from the paper: "MoE-Mamba: Efficient Selective State Space Models with Mixture of Experts" in Pytorch and Ze…☆132Updated this week
- ☆11Aug 17, 2023Updated 3 years ago
- [CVPR 2025] DynRefer: Delving into Region-level Multimodal Tasks via Dynamic Resolution☆59Mar 4, 2025Updated last year
- Simple, minimal implementation of the Mamba SSM in one file of PyTorch.☆2,969Mar 8, 2024Updated 2 years ago
- A Triton Kernel for incorporating Bi-Directionality in Mamba2☆83Dec 18, 2024Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Reading list for research topics in state-space models☆372May 18, 2026Updated 4 months ago
- ☆27Oct 15, 2024Updated last year
- [Mamba-Survey-2024] Paper list for State-Space-Model/Mamba and it's Applications☆754Sep 6, 2026Updated last month
- Awesome Papers related to Mamba.☆1,400Oct 17, 2024Updated last year
- Some preliminary explorations of Mamba's context scaling.☆222Feb 8, 2024Updated 2 years ago
- [ICLR2025] This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision☆92May 30, 2025Updated last year
- [ICML 2023] Stable and Consistent Prediction of 3D Characteristic Orientation vis Invariant Residual Learning☆15May 4, 2026Updated 5 months ago
- More dimensions = More fun☆26Jul 27, 2024Updated 2 years ago
- Implementation of "VL-Mamba: Exploring State Space Models for Multimodal Learning"☆86Mar 21, 2024Updated 2 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- This repository contains the supplementary material for the paper titled: "TRANSFORMER COMPRESSED SENSING VIA GLOBAL IMAGE TOKENS".☆13Dec 20, 2022Updated 3 years ago
- Notes on the Mamba and the S4 model (Mamba: Linear-Time Sequence Modeling with Selective State Spaces)☆187Jan 7, 2024Updated 2 years ago
- Implementation of mamba with rust☆97Mar 9, 2024Updated 2 years ago
- ☆107Mar 9, 2024Updated 2 years ago
- [ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model☆3,908Feb 13, 2025Updated last year
- code for our BMVC 2021 paper "HCV: Hierarchy-Consistency Verification for Incremental Implicitly-Refined Classification"☆15Oct 28, 2022Updated 3 years ago
- ☆52Jan 28, 2024Updated 2 years ago
- Official PyTorch implementation of the Winner Award solution of MIPI 2024 Demosaic for HybridEVS Camera Challenge (CVPR Workshop 2024).☆22Jan 28, 2025Updated last year
- ☆36Nov 22, 2024Updated last year
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- CVPR2024, Semantic-aware SAM for Point-Prompted Instance Segmentation☆37Jan 20, 2025Updated last year
- Official PyTorch Implementation of Efficient and Versatile Robust Fine-Tuning of Zero-shot Models, ECCV 2024☆17Oct 3, 2024Updated 2 years ago
- ☆63Jul 9, 2024Updated 2 years ago
- Swin-UMamba: Mamba-based UNet with ImageNet-based pretraining☆407Mar 19, 2024Updated 2 years ago
- Xmixers: A collection of SOTA efficient token/channel mixers☆29Sep 4, 2025Updated last year
- Integrating Mamba/SSMs with Transformer for Enhanced Long Context and High-Quality Sequence Modeling☆230Oct 2, 2026Updated last week
- A paper list of some recent Mamba-based CV works.☆495Mar 3, 2026Updated 7 months ago