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.
☆107Oct 14, 2025Updated 7 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,460May 3, 2026Updated 3 weeks ago
- VMamba: Visual State Space Models，code is based on mamba☆3,150Mar 7, 2025Updated last year
- Code for paper LocalMamba: Visual State Space Model with Windowed Selective Scan☆282May 6, 2024Updated 2 years ago
- Modified Mamba code to run on CPU☆32Jan 14, 2024Updated 2 years ago
- Official PyTorch Implementation of "The Hidden Attention of Mamba Models"☆232Oct 16, 2025Updated 7 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- [CVPR 2024] Learning SO(3)-Invariant Semantic Correspondence via Local Shape Transform☆14Apr 9, 2026Updated last month
- Implementation of MoE Mamba from the paper: "MoE-Mamba: Efficient Selective State Space Models with Mixture of Experts" in Pytorch and Ze…☆127May 12, 2026Updated last week
- 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,946Mar 8, 2024Updated 2 years ago
- Implementation of paper "CC-Diff: Enhancing Contextual Coherence in Remote Sensing Image Synthesis"☆28Dec 19, 2025Updated 5 months ago
- Reading list for research topics in state-space models☆363Apr 14, 2026Updated last month
- Awesome Papers related to Mamba.☆1,397Oct 17, 2024Updated last year
- ☆26Oct 15, 2024Updated last year
- [Mamba-Survey-2024] Paper list for State-Space-Model/Mamba and it's Applications☆752Jun 28, 2025Updated 10 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- 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☆110Jun 13, 2024Updated last year
- FlashRNN - Fast RNN Kernels with I/O Awareness☆182Oct 20, 2025Updated 7 months ago
- ☆23Jun 17, 2025Updated 11 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
- Implementation of mamba with rust☆94Mar 9, 2024Updated 2 years ago
- Notes on the Mamba and the S4 model (Mamba: Linear-Time Sequence Modeling with Selective State Spaces)☆183Jan 7, 2024Updated 2 years ago
- ☆36Nov 22, 2024Updated last year
- ☆107Mar 9, 2024Updated 2 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- [ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model☆3,869Feb 13, 2025Updated last year
- Official PyTorch Implementation of Efficient and Versatile Robust Fine-Tuning of Zero-shot Models, ECCV 2024☆17Oct 3, 2024Updated last year
- Implementation of a modular, high-performance, and simplistic mamba for high-speed applications☆40Nov 11, 2024Updated last year
- ☆60Jul 9, 2024Updated last year
- Swin-UMamba: Mamba-based UNet with ImageNet-based pretraining☆387Mar 19, 2024Updated 2 years ago
- ☆34Sep 6, 2024Updated last year
- [NeurIPS 2023] Official PyTorch implementation of Binary Radiance Fields☆24Jan 9, 2024Updated 2 years ago
- A paper list of some recent Mamba-based CV works.☆481Mar 3, 2026Updated 2 months ago
- Integrating Mamba/SSMs with Transformer for Enhanced Long Context and High-Quality Sequence Modeling☆220May 11, 2026Updated 2 weeks ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- ✨✨Latest Papers on Vision Mamba and Related Areas☆385Apr 17, 2025Updated last year
- Official Implementation of GENIUS: A Generative Framework for Universal Multimodal Search, CVPR 2025☆55Aug 8, 2025Updated 9 months ago
- Inference of Mamba, Mamba2 and Mamba3 models in pure C☆201Mar 18, 2026Updated 2 months ago
- [NeurIPS 2024] Official repository of MLLA☆375Jul 11, 2025Updated 10 months ago
- Here we will test various linear attention designs.☆62Apr 25, 2024Updated 2 years ago
- collection with description of super-resolution related papers, repositories, datasets, loss functions and etc.☆11Dec 12, 2023Updated 2 years ago
- [NeurIPS 2025] HermesFlow: Seamlessly Closing the Gap in Multimodal Understanding and Generation☆78Sep 19, 2025Updated 8 months ago