search

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

get started with tests #1

Closed jeremiecoullon closed 3 years ago

jeremiecoullon commented 3 years ago

Set up testing (possible with pytest). Start by simply building the various components (diffusions, kernels, samplers) and checking that they return the correct objects (correct shapes/types)

jeremiecoullon commented 3 years ago

Done in the most recent commits