hdembinski / jax-conditional-flows Goto Github PK
View Code? Open in Web Editor NEWThis project forked from smsharma/jax-conditional-flows
Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.
License: MIT License