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

Add Preconditioning to MCLMC #646

Closed reubenharry closed 4 months ago

reubenharry commented 7 months ago

It is often useful to have preconditioning in MCLMC, so that the adaptation algorithm identifies the (approximate, diagonal) covariance of the distribution, and the integrator implicitly works in the mapped space with unit variance in all directions.

This PR adds that. The main change is a parameter std_mat to the esh momentum update.