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

Making sampler composable #157

Closed kazewong closed 7 months ago

kazewong commented 7 months ago

The current workflow of switching between global and local sampler is fixed. This limits the flexibility of dynamic scheduling and early stopping.

It would be useful to wrap both training and sampling into a "strategy block", so the user can composable different training and sampling strategy.

This requires the following work: