blackjax icon indicating copy to clipboard operation
blackjax copied to clipboard

Published 20 hours ago • verified publisher icon blackjax-devs

Rewrite the MNIST MLP example with Flax

Open rlouf opened this issue 2 years ago • 3 comments

https://blackjax-devs.github.io/blackjax/examples/SGMCMC.html

And add SgHMC back!

rlouf avatar Sep 21 '22 17:09 rlouf

Related to #291

zaxtax avatar Sep 21 '22 17:09 zaxtax

Want to do it too?

rlouf avatar Sep 21 '22 18:09 rlouf

@rlouf yea!

zaxtax avatar Sep 21 '22 18:09 zaxtax

Not that you need it, but it should be trivial based on this: https://github.com/blackjax-devs/blackjax/discussions/360#discussioncomment-3756412

rlouf avatar Sep 28 '22 20:09 rlouf

@rlouf I didn't but there are lots of good ideas to take from that!

zaxtax avatar Sep 29 '22 03:09 zaxtax