lindermanlab / ssm-jax

Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend
MIT License
57 stars 7 forks source link

documentation: add dosctrings from ssm/distributions to https://ssm-jax-refactor.readthedocs.io/ #22

Open murphyk opened 2 years ago

murphyk commented 2 years ago

eg for https://github.com/lindermanlab/ssm-jax-refactor/blob/main/ssm/distributions/mvn_block_tridiag.py