An implementation of a family of MCMC methods on implicitly defined manifolds
The api is similar to the one from blackjax but is slightly simplified, and closer to the original MATLAB implementation by @mbrubake .
jax_chmc is still in its early phase. We are actively improving various software components. It is still quite far from ready for everyday use and is made available without any support at the moment.
You can install jax_chmc via pip:
$ pip install jax_chmcTo develop and modify jax_chmc, you need to install
hatch, a tool for Python packaging and
dependency management.
To enter a virtual environment for testing or debugging, you can run:
$ hatch shelljax_chmc uses Pytest for testing. To run the tests, use the following command:
$ hatch run test