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

upgrade packages #68

Closed jeremiecoullon closed 1 year ago

jeremiecoullon commented 1 year ago

Upgrade jax, jaxlib, mypy, black & fix deprecation issues in code