rlouf / mcx

Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.
https://rlouf.github.io/mcx
Apache License 2.0
325 stars 17 forks source link

Online display of divergences #107

Closed rlouf closed 3 years ago

rlouf commented 3 years ago

This PR allows to display the number of divergences that occurred while sampling. Waiting for callbacks to be implemented in #77 to implement as a callback too.