blackjax icon indicating copy to clipboard operation
blackjax copied to clipboard

Published 20 hours ago • verified publisher icon blackjax-devs

Refactor marginal_latent_gaussian.py so it is PyTree compatible

Open rlouf opened this issue 3 years ago • 1 comments

The library has a history of not entirely working with PyTrees, see #216 for instance. We should make sure that it does (adding tests) before the first stable release.

rlouf avatar Jun 29 '22 08:06 rlouf

As discussed in #441 this primarily involves refactoring mgrad. We need some sort of resolution on #77 as well.

rlouf avatar Jan 18 '23 09:01 rlouf