dynamax (pypi): Dynamic State Space Models in JAX.
Keywords: hidden-markov-models|jax|kalman-filter|python|state-space-models