153 use scan to reduce nf compilation time

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 7 months ago

kazewong commented 7 months ago

The forward and backward passes in both realNVP and RQSpline are both using scan.