Bijections & normalizing flows with JAX/NNX
-
Updated
Dec 10, 2025 - Python
Bijections & normalizing flows with JAX/NNX
Surjection layers for density estimation with normalizing flows
Add a description, image, and links to the bijections topic page so that developers can more easily learn about it.
To associate your repository with the bijections topic, visit your repo's landing page and select "manage topics."