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
804 stars 104 forks source link

Add a tutorial for Gaussian Processes with GPJax #436

Open rlouf opened 1 year ago

rlouf commented 1 year ago

We should add a step-by-step example of how to use Blackjax to sample Gaussian Processes models. Part of a series of tutorials that show how Blackjax integrates with the broader JAX ecosystem.

thomaspinder commented 1 year ago

Not sure how/if I can be helpful here, but please let me know if there's anything I can help with.

rlouf commented 1 year ago

Thanks! We're looking for something simple like your classification example (but only with the inference part). It's essentially to signal potential users that they can use Blackjax with GPJax easily.

I also noticed there is some variational inference functionality in your library and I wonder if at least part of it could be handled by Blackjax now that we're also moving in the VI direction.