Popular GitHub Repositories for State Space Models

State Space Models library in JAX

Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.

[ACM MM'24 Oral] RainMamba: Enhanced Locality Learning with State Space Models for Video Deraining

Official implementation of our ECCV paper "StretchBEV: Stretching Future Instance Prediction Spatially and Temporally"

List of papers related to State Space Models (Mamba) in Vision.

[NeurIPS 2024] Official implementation of the paper "MambaLRP: Explaining Selective State Space Sequence Models".

Official Pytorch implementation of NeuralWalker

Official implementation of DenoMamba: A fused state-space model for low-dose CT denoising

Official implementation of MambaRoll: A Physics-Driven Autoregressive State Space Model for Medical Image Reconstruction (https://arxiv.org/abs/2412.09331)

Official implementation of the CBF-SSM model

Variational Filtering via Wasserstein Gradient Flow

Switching linear dynamical systems (SLDS) models in JAX

Stock Movement Prediction Using Mamba And Ensemble Learning

self-driving using end-to-end state space models

A comprehensive learning guide for understanding S4 (Structured State Space Sequence Model) and Mamba architectures, from fundamentals through implementation, with curated resources and hands-on examples.

Comparing length generalization performance for Mamba, attention-based transformers and Mamba-attention hybrids