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 8 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,471May 3, 2026Updated 2 months ago
- Code for paper LocalMamba: Visual State Space Model with Windowed Selective Scan☆282May 6, 2024Updated 2 years ago
- ☆15Oct 14, 2025Updated 8 months ago
- Official PyTorch Implementation of "The Hidden Attention of Mamba Models"☆232Oct 16, 2025Updated 8 months ago
- Implementation of MoE Mamba from the paper: "MoE-Mamba: Efficient Selective State Space Models with Mixture of Experts" in Pytorch and Ze…☆132Jun 22, 2026Updated last week
- 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.
- 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,960Mar 8, 2024Updated 2 years ago
- Reading list for research topics in state-space models☆365May 18, 2026Updated last month
- Awesome Papers related to Mamba.☆1,398Oct 17, 2024Updated last year
- The official implementation of BiViT: Extremely Compressed Binary Vision Transformers☆16Jun 18, 2023Updated 3 years ago
- [Mamba-Survey-2024] Paper list for State-Space-Model/Mamba and it's Applications☆752Jun 28, 2025Updated last year
- Some preliminary explorations of Mamba's context scaling.☆220Feb 8, 2024Updated 2 years ago
- [ICLR2025] This repository is the official implementation of our Autoregressive Pretraining with Mamba in Vision☆90May 30, 2025Updated last year
- More dimensions = More fun☆26Jul 27, 2024Updated last year
- 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.
- [NeurIPS2024 Spotlight] The official implementation of MambaTree: Tree Topology is All You Need in State Space Model☆113Jun 13, 2024Updated 2 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
- FlashRNN - Fast RNN Kernels with I/O Awareness☆185Oct 20, 2025Updated 8 months ago
- ☆23Jun 17, 2025Updated last year
- Official PyTorch implementation of the Winner Award solution of MIPI 2024 Demosaic for HybridEVS Camera Challenge (CVPR Workshop 2024).☆19Jan 28, 2025Updated last year
- Implementation of mamba with rust☆96Mar 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
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- ☆52Jan 28, 2024Updated 2 years ago
- [ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model☆3,885Feb 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☆41Nov 11, 2024Updated last year
- ☆61Jul 9, 2024Updated last year
- Swin-UMamba: Mamba-based UNet with ImageNet-based pretraining☆389Mar 19, 2024Updated 2 years ago
- TIP: Bi-directional Exponential Angular Triplet Loss for RGB-Infrared Person Re-Identification☆21Mar 29, 2021Updated 5 years ago
- Multi-Resolution Neural Networks☆22Apr 13, 2026Updated 2 months ago
- A paper list of some recent Mamba-based CV works.☆489Mar 3, 2026Updated 4 months 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.
- [NeurIPS 2023] Official PyTorch implementation of Binary Radiance Fields☆23Jan 9, 2024Updated 2 years ago
- Integrating Mamba/SSMs with Transformer for Enhanced Long Context and High-Quality Sequence Modeling☆225Jun 22, 2026Updated last week
- ✨✨Latest Papers on Vision Mamba and Related Areas☆385Apr 17, 2025Updated last year
- ☆11Feb 9, 2026Updated 4 months ago
- Official Implementation of GENIUS: A Generative Framework for Universal Multimodal Search, CVPR 2025☆55Aug 8, 2025Updated 10 months ago
- [NeurIPS 2024] Official repository of MLLA☆374Jul 11, 2025Updated 11 months ago
- Here we will test various linear attention designs.☆62Apr 25, 2024Updated 2 years ago