search

kazewong / flowMC

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

158 put training loop into nf class #160

Closed kazewong closed 5 months ago

kazewong commented 5 months ago

Move the training loop into the normalizing flow for a cleaner API