jax-ml / bayeux

State of the art inference for your bayesian models.
https://jax-ml.github.io/bayeux/
Apache License 2.0
135 stars 5 forks source link

Add meads and chees samplers from blackjax. #11

Closed copybara-service[bot] closed 8 months ago

copybara-service[bot] commented 8 months ago

Add meads and chees samplers from blackjax.

These samplers pass the current MCMC tests (which just require sampling from a 1d gaussian), but do poorly enough on actual problems that there is probably a bug somewhere. I think this is a blackjax problem, but exposing these methods may help track down where the problem is.