blackjax-devs / blackjax

BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.
https://blackjax-devs.github.io/blackjax/
Apache License 2.0
806 stars 105 forks source link

Adjusted MCLMC #676

Open reubenharry opened 4 months ago

reubenharry commented 4 months ago

Presentation of the new sampler

MCLMC is an unadjusted algorithm, in the sense that no MH step is performed, and it is unbiased. As it turns out, it is possible to calculate the MH term, so we can implement an adjusted version.

How does it compare to other algorithms in blackjax?

Similar to MCLMC, but with adjustment. In most cases, it will be slower to converge (especially in high dimensions), but our tests suggest that it is generally faster than NUTS.

Where does it fit in blackjax

Next to HMC and NUTS. It may be possible even to share some code.

Are you willing to open a PR?

Yes, forthcoming.