README.md
setup.py
examples/__init__.py
examples/cifar10.py
examples/debug.py
examples/flow_matching.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