Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).
☆132Oct 18, 2024Updated last year
Alternatives and similar repositories for mamba-tiny
Users that are interested in mamba-tiny are comparing it to the libraries listed below
Sorting:
- Notes on the Mamba and the S4 model (Mamba: Linear-Time Sequence Modeling with Selective State Spaces)☆180Jan 7, 2024Updated 2 years ago
- Simple, minimal implementation of the Mamba SSM in one file of PyTorch.☆2,929Mar 8, 2024Updated 2 years ago
- A simple and efficient Mamba implementation in pure PyTorch and MLX.☆1,442Jan 26, 2026Updated last month
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆15Aug 31, 2023Updated 2 years ago
- PyTorch implementation of Structured State Space for Sequence Modeling (S4), based on Annotated S4.☆89Mar 1, 2024Updated 2 years ago
- Out-of-Distribution detection for segmentation data, using deep nearest neighbors. Presented at the ICCV23 workshop on Uncertainty Quanti…☆15Oct 9, 2023Updated 2 years ago
- Target speaker automatic speech recognition (TS-ASR)☆12Oct 14, 2023Updated 2 years ago
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆18Mar 15, 2024Updated 2 years ago
- Mamba SSM architecture☆17,524Updated this week
- My personal toolbox for doing datascience (especially deep learning) in python.☆18Mar 21, 2020Updated 6 years ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆93Jan 25, 2024Updated 2 years ago
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆98Dec 5, 2024Updated last year
- ☆40Jan 5, 2024Updated 2 years ago
- Generative Modeling via Drifting in MLX☆42Feb 6, 2026Updated last month
- Annotated version of the Mamba paper☆499Feb 27, 2024Updated 2 years ago
- Evaluating the Mamba architecture on the Othello game☆49Apr 25, 2024Updated last year
- Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.☆54Apr 12, 2024Updated last year
- ☆22Apr 22, 2024Updated last year
- GP Sinkhorn Implementation, paper: https://www.mdpi.com/1099-4300/23/9/1134☆23May 1, 2022Updated 3 years ago
- Official PyTorch Implementation of "The Hidden Attention of Mamba Models"☆231Oct 16, 2025Updated 5 months ago
- Simple voice activity detection (VAD) algorithm in Python☆15Aug 10, 2023Updated 2 years ago
- [NeurIPS 2024] Official implementation of the paper "MambaLRP: Explaining Selective State Space Sequence Models" 🐍☆45Nov 6, 2024Updated last year
- ☆18Oct 26, 2024Updated last year
- Implementation of a modular, high-performance, and simplistic mamba for high-speed applications☆40Nov 11, 2024Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆83May 17, 2024Updated last year
- Paper implementation☆50Apr 8, 2025Updated 11 months ago
- ☆11Oct 11, 2023Updated 2 years ago
- Preprocessed data of SignDiff: Learning Diffusion Models for American Sign Language Production☆17May 1, 2025Updated 10 months ago
- ☆15Jul 24, 2022Updated 3 years ago
- Code for Fooling Contrastive Language-Image Pre-trainined Models with CLIPMasterPrints☆15Jan 25, 2026Updated last month
- ☆56Apr 28, 2025Updated 10 months ago
- Transformer from scratch with einsum method☆11Jul 8, 2021Updated 4 years ago
- Collect papers about Mamba (a selective state space model).☆14Aug 6, 2024Updated last year
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆62Sep 3, 2025Updated 6 months ago
- Python wrapper for the DPMMSubClusterStreaming.jl Julia package.☆14Sep 9, 2022Updated 3 years ago
- Examples of using PyTorch hooks, as covered in my YouTube tutorial video.☆38Oct 25, 2023Updated 2 years ago
- Official repository for "Boosting Adversarial Transferability using Dynamic Cues " (ICLR 2023)☆20Aug 24, 2023Updated 2 years ago
- RWKV-X is a Linear Complexity Hybrid Language Model based on the RWKV architecture, integrating Sparse Attention to improve the model's l…☆56Updated this week
- Computer Vision Course 2024-01☆12Apr 19, 2024Updated last year