search

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
794 stars 102 forks source link

Refactor `Pathfinder` #465

Open rlouf opened 1 year ago

rlouf commented 1 year ago

We should update Pathfinder’s API to match Mean field VI’s.

junpenglao commented 1 year ago

Related: https://github.com/blackjax-devs/blackjax/issues/387

conorhassan commented 1 year ago

I am keen to have a crack at this issue if it is still up for grabs. I understand that the API for mean field is still in flux a little. Is there a sketch outline for what people think the API should like, or should I go off the current main branch?

rlouf commented 1 year ago

I think the high-level MFVI API is pretty much what we'd want it to look like

conorhassan commented 1 year ago

Sounds good. I'm keen to give this a go.