Code Repository for Liquid Time-Constant Networks (LTCs)

State Space Models library in JAX

Liquid Structural State-Space Models

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

Mambular is a Python package that brings the power of Mamba architectures to tabular data, offering a suite of deep learning models for regression, classification, and distributional regression tasks. This includes models like Mambular, FT-Transformer, TabTransformer and tabular ResNets.

[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

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

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

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

Official repository for Mamba-based Segmentation Model for Speaker Diarization

Official Pytorch implementation of NeuralWalker

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

Gradient-informed particle MCMC methods

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

Second-order iterated smoothing algorithms for state estimation

Variational Filtering via Wasserstein Gradient Flow

Official implementation of the CBF-SSM model

A Python package to demonstrate ideas from nonlinear dynamical systems toward game theory, neural network models of associative memory, and nonlinear state space models.