GitHub repository: probml/dynamax
Primary language: Python
Description: A Python package for probabilistic state space modeling with JAX