search

kazewong / flowMC

Normalizing-flow enhanced sampling package for probabilistic inference in Jax
https://flowmc.readthedocs.io/en/main/
MIT License
200 stars 23 forks source link

157 making sampler composable #161

Closed kazewong closed 7 months ago

kazewong commented 7 months ago

There is the basic infrastructure for composable strategy now.

Fix a bunch of bugs in the normalizing flow interface that were introduced in #160. The performance seems to be okay now, but need to check with community example to make sure that is the case