Simple, minimal implementation of the Mamba SSM in one file of PyTorch.
☆2,970Mar 8, 2024Updated 2 years ago
Alternatives and similar repositories for mamba-minimal
Users that are interested in mamba-minimal are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Mamba SSM architecture☆18,873Jul 22, 2026Updated 2 months ago
- A simple and efficient Mamba implementation in pure PyTorch and MLX.☆1,481May 3, 2026Updated 4 months ago
- [ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model☆3,905Feb 13, 2025Updated last year
- Annotated version of the Mamba paper☆504Feb 27, 2024Updated 2 years ago
- Structured state space sequence models☆2,925Jul 17, 2024Updated 2 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Mamba-Chat: A chat LLM based on the state-space model architecture 🐍☆942Mar 3, 2024Updated 2 years ago
- Awesome Papers related to Mamba.☆1,400Oct 17, 2024Updated last year
- VMamba: Visual State Space Models，code is based on mamba☆3,233Mar 7, 2025Updated last year
- High-speed Large Language Model Serving for Local Deployment☆9,810May 11, 2026Updated 4 months ago
- 🚀 Efficient implementations for emerging model architectures☆5,795Updated this week
- Collection of papers on state-space models☆620Aug 10, 2026Updated last month
- RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable)…☆14,728Updated this week
- Fast and memory-efficient exact attention☆25,022Updated this week
- ☆36Nov 22, 2024Updated last year
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Minimal Mamba-2 implementation in PyTorch☆258Jun 17, 2024Updated 2 years ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆258Jun 6, 2025Updated last year
- Some preliminary explorations of Mamba's context scaling.☆222Feb 8, 2024Updated 2 years ago
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆177Jan 30, 2025Updated last year
- Understand and test language model architectures on synthetic tasks.☆287Mar 22, 2026Updated 6 months ago
- [ICLR 2025] Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling☆979Nov 16, 2025Updated 10 months ago
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆6,256Aug 22, 2025Updated last year
- The TinyLlama project is an open endeavor to pretrain a 1.1B Llama model on 3 trillion tokens.☆9,013May 3, 2024Updated 2 years ago
- Implementation of https://srush.github.io/annotated-s4☆519Jun 20, 2025Updated last year
- 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.
- Integrating Mamba/SSMs with Transformer for Enhanced Long Context and High-Quality Sequence Modeling☆229Updated this week
- Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.☆57Apr 12, 2024Updated 2 years ago
- Tile primitives for speedy kernels☆3,730Sep 12, 2026Updated 2 weeks ago
- Kolmogorov Arnold Networks☆16,336Jan 19, 2025Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆93Jan 25, 2024Updated 2 years ago
- ☆4,123Apr 15, 2026Updated 5 months ago
- Official repository of the xLSTM.☆2,205Sep 7, 2026Updated 2 weeks ago
- Open weights language model from Google DeepMind, based on Griffin.☆685Feb 6, 2026Updated 7 months ago
- Schedule-Free Optimization in PyTorch☆2,322Jul 28, 2026Updated last month
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).☆135Oct 18, 2024Updated last year
- [ICLR 2024] Efficient Streaming Language Models with Attention Sinks☆7,268Jul 11, 2024Updated 2 years ago
- [NeurIPS 2023 spotlight] Official implementation of HGRN in our NeurIPS 2023 paper - Hierarchically Gated Recurrent Neural Network for Se…☆68Apr 24, 2024Updated 2 years ago
- Tools for merging pretrained large language models.☆7,376Sep 12, 2026Updated 2 weeks ago
- A PyTorch native platform for training generative AI models☆5,771Updated this week
- Development repository for the Triton language and compiler☆20,247Updated this week
- Implementation of the Mamba SSM with hf_integration.☆56Aug 31, 2024Updated 2 years ago