Get rid of random_key_set

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

Get rid of random_key_set #144

Closed kazewong closed 7 months ago

kazewong commented 9 months ago

Mentioned in #141, the random_key_set variable in flowMC is probably an anti-pattern. It introduce unnecessary complication in initializing the sampler.

Would be nice to remove this, and follow jax.random.PRNGKey practice closer.