State Space Model Architectures ML Frameworks

Implementations and applications of Mamba and selective state-space models across modalities (vision, audio, medical imaging, video). Does NOT include general sequence models, transformers, or non-SSM architectures.

There are 22 state space model architectures frameworks tracked. The highest-rated is kyegomez/VisionMamba at 43/100 with 482 stars.

Get all 22 projects as JSON

curl "https://pt-edge.onrender.com/api/v1/datasets/quality?domain=ml-frameworks&subcategory=state-space-model-architectures&limit=20"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

#	Framework	Score	Tier	Stars	Language
1	kyegomez/VisionMamba Implementation of Vision Mamba from the paper: "Vision Mamba: Efficient...	43	Emerging	482	Python
2	kkakkkka/MambaTalk [NeurlPS-2024] The official code of MambaTalk: Efficient Holistic Gesture...	38	Emerging	76	Python
3	SiavashShams/ssamba [SLT'24] The official implementation of SSAMBA: Self-Supervised Audio...	37	Emerging	134	Python
4	kaistmm/Audio-Mamba-AuM Official Implementation of the work "Audio Mamba: Bidirectional State Space...	34	Emerging	167	Python
5	FarnoushRJ/MambaLRP [NeurIPS 2024] Official implementation of the paper "MambaLRP: Explaining...	33	Emerging	45	Python
6	zs1314/SkinMamba 【ACCVW2025 Oral】Offical Pytorch Code for "SkinMamba: A Precision Skin Lesion...	30	Emerging	51	Python
7	niyiyu/DAS-reconstruction Wavefield Reconstruction of Distributed Acoustic Sensing with Machine Learning	30	Emerging	16	Jupyter Notebook
8	YichuXu/MambaMoE [Information Fusion 2025] MambaMoE: Mixture-of-Spectral-Spatial-Experts...	26	Experimental	33	Python
9	xmaptools/XMapTools_Developers This developer repository contains the latest version of the XMapTools...	26	Experimental	6	MATLAB
10	YichuXu/PHDMamba [IEEE GRSL 2025] PHDMamba: Progressive Hybrid Mamba for Hyperspectral Image...	24	Experimental	13	Python
11	icon-lab/MambaRoll Official implementation of MambaRoll: A Physics-Driven Autoregressive State...	22	Experimental	19	Python
12	m1dsolo/EchoCardMAE EchoCardMAE: Video Masked Auto-Encoders Customized for Echocardiography	21	Experimental	9	Python
13	zs1314/MambaMIC 【ICME2025 Oral】Offical Pytorch Code for "MambaMIC: An Efficient Baseline for...	21	Experimental	8	Python
14	cherise215/AdvBias [MICCAI 2020 Oral] Realistic Adversarial Data Augmentation for MR Image Segmentation	20	Experimental	16	Jupyter Notebook
15	ChiShengChen/ResVMamba The official repository implement of Res-VMamba: Fine-Grained Food Category...	19	Experimental	83	Python
16	maklachur/Mamba-in-Computer-Vision Mamba in Vision: A Comprehensive Survey of Techniques and Applications	19	Experimental	137	—
17	cocoalex00/Mamba2D Official PyTorch Implementation of Mamba2D: A Natively Multi-Dimensional...	18	Experimental	12	Python
18	OliverRensu/ARM [ICLR2025] This repository is the official implementation of our...	18	Experimental	90	Python
19	mhbenabda/deep-learning-optoacoustic-reconstruction End-to-end pipeline for optoacoustic image reconstruction using DAS, deep...	17	Experimental	2	Jupyter Notebook
20	ghostperpper007/small_programming_model A from-scratch Python code model with GNN-based structure encoding,...	14	Experimental	—	Python
21	Efficient-Scalable-Machine-Learning/event-based-mamba Building on our prior work on event-based state-space models, we developed...	14	Experimental	3	Python
22	xmindflow/SSL-contrastive [ISBI 2024] Leveraging Unlabeled Data for 3D Medical Image Segmentation...	12	Experimental	25	Python