jax-flows

Depends On

• jax (jax)
• jaxlib (jaxlib)
• numpy (numpy)
• scipy (scipy)
• sklearn (sklearn)