State Space Model Python Packages

pyhgf

PyHGF: A neural network library for predictive coding

replay-trajectory-classification

State space models for decoding hippocampal trajectories and determining their type using sorted or clusterless data

tf2ss

Pure Python MIMO transfer function to state-space conversion. Easy-to-install alternative to SLYCOT with MATLAB-consistent results. No Fortran compiler required! 🚀

lattice-dsp

Stable IIR lattice filters, adaptive recursive DSP, and matrix/MIMO lattice experiments for Python.

replay-classification

State space models for categorization of replay content from multiunit spiking activity. Deng et al. 2016

unsupervised-multimodal-trajectory-modeling

We use EM for a mixture of state space models to perform unsupervised clustering of short trajectories.

long-range-models

Simple implementations of long-range sequence models (LRU, S5, S4, and more).

marmo

Markov-like processes as one recursion over a belief algebra and a semiring: HMM, HSMM, Kalman, and particle inference behind a decidable engine dispatch.

dynaris

A JAX-powered Python library for Dynamic Linear Models (DLMs)

s0-tuning

Tune the initial recurrent state of hybrid models. Zero inference overhead.

alloy-mlx

Hybrid SSM-Attention language model on Apple Silicon with MLX

statecraft

Store, manage and remix states for SSMs and other Stateful models