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

134 add precommit script and fix formatting #151

Closed kazewong closed 7 months ago

kazewong commented 7 months ago

Fixed most of the typing hints in local samplers and normalizing flow modules.

Since we are going to rework the main Sampler class, the typing hints will be added there incrementally.