State Space Models in JAX.
Authors: Peter Chang, Adrien Corenflos, Gerardo Duran-Martin, Giles Harper-Donnelly, Aleyna Kara, Scott Linderman, Kevin Murphy, Colin Schlager, et al.
MIT License. 2022
To run a specific demo, do something like this
python ssm_jax/hmm/demos/gaussian_hmm_2d.py
To run all the tests, do this
pytest ssm_jax
To run a specific test, do something like this
pytest ssm_jax/hmm/inference_test.py
pytest ssm_jax/hmm/demos/demos_test.py
To install in colab, do this
!git clone https://github.com/probml/ssm-jax.git
%cd ssm-jax
!pip install -e .
To install black, do this (quotes are mandatory for zsh
)
pip install -U 'black[jupyter]'
Related libraries: