Official PyTorch Implementation of "The Hidden Attention of Mamba Models"
☆234Oct 16, 2025Updated 9 months ago
Alternatives and similar repositories for HiddenMambaAttn
Users that are interested in HiddenMambaAttn are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- [ICLR 2025] Official Code Release for Explaining Modern Gated-Linear RNNs via a Unified Implicit Attention Formulation☆50Mar 1, 2025Updated last year
- [NeurIPS 2024] Official implementation of the paper "MambaLRP: Explaining Selective State Space Sequence Models" 🐍☆47Nov 6, 2024Updated last year
- Implementation of MambaFormer in Pytorch ++ Zeta from the paper: "Can Mamba Learn How to Learn? A Comparative Study on In-Context Learnin…☆21Updated this week
- Some preliminary explorations of Mamba's context scaling.☆220Feb 8, 2024Updated 2 years ago
- [Mamba-Survey-2024] Paper list for State-Space-Model/Mamba and it's Applications☆754Updated this week
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- More dimensions = More fun☆26Jul 27, 2024Updated last year
- Reading list for research topics in state-space models☆366May 18, 2026Updated 2 months ago
- DeciMamba: Exploring the Length Extrapolation Potential of Mamba (ICLR 2025)☆32Apr 9, 2025Updated last year
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆18Mar 15, 2024Updated 2 years ago
- Code for paper LocalMamba: Visual State Space Model with Windowed Selective Scan☆283May 6, 2024Updated 2 years ago
- ☆40Feb 26, 2024Updated 2 years ago
- Awesome Papers related to Mamba.☆1,399Oct 17, 2024Updated last year
- A simple and efficient Mamba implementation in pure PyTorch and MLX.☆1,471May 3, 2026Updated 2 months ago
- A repository for DenseSSMs☆90Apr 11, 2024Updated 2 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆174Jan 30, 2025Updated last year
- Annotated version of the Mamba paper☆501Feb 27, 2024Updated 2 years ago
- The accompanying code for "Simplifying and Understanding State Space Models with Diagonal Linear RNNs" (Ankit Gupta, Harsh Mehta, Jonatha…☆23Dec 30, 2022Updated 3 years ago
- VMamba: Visual State Space Models，code is based on mamba☆3,206Mar 7, 2025Updated last year
- ☆35Apr 12, 2024Updated 2 years ago
- ☆11Oct 11, 2023Updated 2 years ago
- A single repo with all scripts and utils to train / fine-tune the Mamba model with or without FIM☆62Apr 8, 2024Updated 2 years ago
- [ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model☆3,889Feb 13, 2025Updated last year
- Official Code Repository for the paper "Key-value memory in the brain"☆32Feb 25, 2025Updated last year
- 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☆83Dec 18, 2024Updated last year
- Curse-of-memory phenomenon of RNNs in sequence modelling☆19May 8, 2025Updated last year
- A curated collection of papers, tutorials, videos, and other valuable resources related to Mamba.☆750Jun 11, 2026Updated last month
- An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivatio…☆109Oct 14, 2025Updated 9 months ago
- [ICCV2025] Introduce Mamba2 to Vision.☆190Oct 29, 2025Updated 8 months ago
- A PyTorch wrapper of parallel exclusive scan in CUDA☆12May 25, 2023Updated 3 years ago
- Mamba SSM architecture☆18,636Jul 7, 2026Updated 2 weeks ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆256Jun 6, 2025Updated last year
- ☆52Jan 28, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Implementation of MoE Mamba from the paper: "MoE-Mamba: Efficient Selective State Space Models with Mixture of Experts" in Pytorch and Ze…☆132Jul 13, 2026Updated last week
- [IEEE TCSVT] Vivim: a Video Vision Mamba for Medical Video Segmentation☆192Jun 12, 2025Updated last year
- This is an efficient cuda implementation of 2D depthwise convolution for large kernel, it can be used in Pytorch deep learning framework.☆12Sep 28, 2023Updated 2 years ago
- Implementation of MambaByte in "MambaByte: Token-free Selective State Space Model" in Pytorch and Zeta☆128Updated this week
- Computation-Efficient Era: A Comprehensive Survey of State Space Models in Medical Image Analysis☆268Jul 22, 2025Updated 11 months ago
- ☆204May 27, 2024Updated 2 years ago
- ☆49Mar 31, 2024Updated 2 years ago