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

torch-dimensions

Torch for Multidimensional Compatibility and Easy to Go protocols for both CUDA and MPS. (SSMs are also provided probes for Apple MPS training/inferencing)

unsupervised-multimodal-trajectory-modeling

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

dynaris

A JAX-powered library for state-space models

statecraft

Statecraft - Load, store and remix states for SSMs, Mamba and Stateful models

alloy-mlx

Hybrid SSM-Attention language model on Apple Silicon with MLX — interleaving Mamba-2 and Transformer for efficient inference

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.

long-range-models

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