jeremiecoullon / SGMCMCJax

Lightweight library of stochastic gradient MCMC algorithms written in JAX.
https://sgmcmcjax.readthedocs.io/en/latest/index.html
Apache License 2.0
95 stars 8 forks source link

Add docstrings and link to documentation #3

Closed jeremiecoullon closed 2 years ago

jeremiecoullon commented 3 years ago

Add docstrings to all the functions and link this up to the documentation