Code Repository for Liquid Time-Constant Networks (LTCs)

A Python package for probabilistic state space modeling with JAX

Liquid Structural State-Space Models

DeepTab is a Python package that simplifies tabular deep learning by providing a suite of models for regression, classification, and distributional regression tasks. It includes models such as Mambular, TabM, FT-Transformer, TabulaRNN, TabTransformer, and tabular ResNets.

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

[CVPR'24 Spotlight] The official implementation of "State Space Models for Event Cameras"

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

Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024

A toolkit for developing foundation models using Electronic Health Record (EHR) data.

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

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

Official repository for Mamba-based Segmentation Model for Speaker Diarization

Official Pytorch implementation of NeuralWalker (ICLR 2025)

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

[CVPR'25 Highlight] The official implementation of "GG-SSMs: Graph-Generating State Space Models"

linax is a light weight collection of state space models implemented in JAX ⚡️

Temporal Neural Networks

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

Simulates the dynamics of a Reaction Wheel Inverted Pendulum with python.

Gradient-informed particle MCMC methods