README.md
setup.py
generax/__init__.py
generax/distributions.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/flow/__init__.py
generax/flow/affine.py
generax/flow/base.py
generax/flow/coupling.py
generax/flow/models.py
generax/flow/reshape.py
generax/flow/spline.py
generax/nn/__init__.py
generax/nn/flat_net.py
generax/nn/neural_ode.py
generax/nn/resnet_1d.py
generax/nn/time_condition.py
generax/nn/unet.py
generax/nn/util.py
generax/training/__init__.py
generax/training/trainer.py
generax/training/max_likelihood/__init__.py
generax/training/max_likelihood/ml.py
notebooks/__init__.py