165 add probability floor to normalizing flow model -> updating Adam optimizer to optimize with constraint

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

Closed kazewong closed 6 months ago

kazewong commented 6 months ago

In this PR, we actually decided not to move forward with the probability floor since that seems to make the training and sampling quite unstable.

On the other hand, we update the Adam optimization interface such that it can perform constrained optimization.