FOX (FlOws in JAX)

22 May 2021 · 1 min read

Developing a Python package with standard normalizing flow implementations that is based on JAX and Flax.