README.md
setup.py
examples/__init__.py
examples/celeba.py
examples/cifar10.py
examples/cnf_max_likelihood.py
examples/debug.py
examples/flow_matching.py
examples/mnist.py
examples/multisample_flow_matching.py
examples/score_matching.py
examples/spline_max_likelihood.py
generax/__init__.py
generax/trainer.py
generax.egg-info/PKG-INFO
generax.egg-info/SOURCES.txt
generax.egg-info/dependency_links.txt
generax.egg-info/requires.txt
generax.egg-info/top_level.txt
generax/distributions/__init__.py
generax/distributions/base.py
generax/distributions/coupling.py
generax/distributions/flow_models.py
generax/flows/__init__.py
generax/flows/affine.py
generax/flows/base.py
generax/flows/coupling.py
generax/flows/ffjord.py
generax/flows/logistic_cdf_mixture_logit.py
generax/flows/models.py
generax/flows/nonlinearities.py
generax/flows/reshape.py
generax/flows/spline.py
generax/nn/__init__.py
generax/nn/grad_wrapper.py
generax/nn/input_convex.py
generax/nn/layers.py
generax/nn/lipschitz.py
generax/nn/neural_ode.py
generax/nn/resnet.py
generax/nn/resnet_blocks.py
generax/nn/time_condition.py
generax/nn/unet.py
generax/tests/__init__.py
generax/tests/flow_tests.py
generax/util/__init__.py
generax/util/iterative.py
generax/util/misc.py
misc/__init__.py
misc/cnf_neals_funnel.py
misc/cnf_vi.py
misc/realnvp_vi.py
misc/vi/__init__.py
misc/vi/cnf_gmm.py