/jax-conditional-flows

Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.

Primary LanguageJupyter NotebookMIT LicenseMIT

Stargazers