├── LICENSE ├── README.md ├── conda_environment.yml ├── setup.py ├── snf ├── __init__.py ├── cli.py ├── datasets │ ├── __init__.py │ ├── cifar10.py │ ├── imagenet.py │ ├── mnist.py │ └── toy_density_data.py ├── experiments │ ├── __init__.py │ ├── conv1x1_glow_cifar.py │ ├── conv1x1_glow_imagenet.py │ ├── conv1x1_glow_mnist.py │ ├── emerging_cnn_mnist.py │ ├── exact_cnn_mnist.py │ ├── exact_fc_mnist.py │ ├── exponential_cnn_mnist.py │ ├── geco_selfnorm_glow_mnist.py │ ├── selfnorm_cnn_mnist.py │ ├── selfnorm_fc_mnist.py │ ├── selfnorm_glow_cifar.py │ ├── selfnorm_glow_imagenet.py │ ├── selfnorm_glow_mnist.py │ └── snf_timescaling.py ├── layers │ ├── __init__.py │ ├── activations.py │ ├── actnorm.py │ ├── conv1x1.py │ ├── convexp │ │ ├── __init__.py │ │ ├── convexp_module.py │ │ ├── functional.py │ │ └── spectral.py │ ├── coupling.py │ ├── dequantize.py │ ├── distributions │ │ ├── __init__.py │ │ ├── gaussian.py │ │ └── uniform.py │ ├── emerging │ │ ├── __init__.py │ │ ├── emerging_module.py │ │ ├── inverse_op_cython.cpython-37m-x86_64-linux-gnu.so │ │ ├── inverse_op_cython.pyx │ │ ├── inverse_op_cython.pyxbld │ │ ├── inverse_triang_conv.py │ │ └── masks.py │ ├── flowlayer.py │ ├── flowsequential.py │ ├── normalize.py │ ├── selfnorm.py │ ├── splines │ │ ├── __init__.py │ │ └── rational_quadratic.py │ ├── splitprior.py │ ├── squeeze.py │ └── transforms.py ├── train │ ├── __init__.py │ ├── datatransforms.py │ ├── experiment.py │ ├── losses.py │ └── statsrecorder.py └── utils │ ├── __init__.py │ ├── convbackward │ ├── __init__.py │ ├── conv2d_backward.cpp │ └── torch_cpp_extensions │ │ ├── .ninja_deps │ │ ├── .ninja_log │ │ ├── .placeholder │ │ ├── build.ninja │ │ ├── conv2d_backward.o │ │ └── conv2d_backward.so │ └── toeplitz.py └── tests └── snf └── test_layers.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/README.md -------------------------------------------------------------------------------- /conda_environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/conda_environment.yml -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/setup.py -------------------------------------------------------------------------------- /snf/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /snf/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/cli.py -------------------------------------------------------------------------------- /snf/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /snf/datasets/cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/datasets/cifar10.py -------------------------------------------------------------------------------- /snf/datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/datasets/imagenet.py -------------------------------------------------------------------------------- /snf/datasets/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/datasets/mnist.py -------------------------------------------------------------------------------- /snf/datasets/toy_density_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/datasets/toy_density_data.py -------------------------------------------------------------------------------- /snf/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /snf/experiments/conv1x1_glow_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/experiments/conv1x1_glow_cifar.py -------------------------------------------------------------------------------- /snf/experiments/conv1x1_glow_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/experiments/conv1x1_glow_imagenet.py -------------------------------------------------------------------------------- /snf/experiments/conv1x1_glow_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/experiments/conv1x1_glow_mnist.py -------------------------------------------------------------------------------- /snf/experiments/emerging_cnn_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/experiments/emerging_cnn_mnist.py -------------------------------------------------------------------------------- /snf/experiments/exact_cnn_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/experiments/exact_cnn_mnist.py -------------------------------------------------------------------------------- /snf/experiments/exact_fc_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/experiments/exact_fc_mnist.py -------------------------------------------------------------------------------- /snf/experiments/exponential_cnn_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/experiments/exponential_cnn_mnist.py -------------------------------------------------------------------------------- /snf/experiments/geco_selfnorm_glow_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/experiments/geco_selfnorm_glow_mnist.py -------------------------------------------------------------------------------- /snf/experiments/selfnorm_cnn_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/experiments/selfnorm_cnn_mnist.py -------------------------------------------------------------------------------- /snf/experiments/selfnorm_fc_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/experiments/selfnorm_fc_mnist.py -------------------------------------------------------------------------------- /snf/experiments/selfnorm_glow_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/experiments/selfnorm_glow_cifar.py -------------------------------------------------------------------------------- /snf/experiments/selfnorm_glow_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/experiments/selfnorm_glow_imagenet.py -------------------------------------------------------------------------------- /snf/experiments/selfnorm_glow_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/experiments/selfnorm_glow_mnist.py -------------------------------------------------------------------------------- /snf/experiments/snf_timescaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/experiments/snf_timescaling.py -------------------------------------------------------------------------------- /snf/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/__init__.py -------------------------------------------------------------------------------- /snf/layers/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/activations.py -------------------------------------------------------------------------------- /snf/layers/actnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/actnorm.py -------------------------------------------------------------------------------- /snf/layers/conv1x1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/conv1x1.py -------------------------------------------------------------------------------- /snf/layers/convexp/__init__.py: -------------------------------------------------------------------------------- 1 | from . import functional 2 | -------------------------------------------------------------------------------- /snf/layers/convexp/convexp_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/convexp/convexp_module.py -------------------------------------------------------------------------------- /snf/layers/convexp/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/convexp/functional.py -------------------------------------------------------------------------------- /snf/layers/convexp/spectral.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/convexp/spectral.py -------------------------------------------------------------------------------- /snf/layers/coupling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/coupling.py -------------------------------------------------------------------------------- /snf/layers/dequantize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/dequantize.py -------------------------------------------------------------------------------- /snf/layers/distributions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/distributions/__init__.py -------------------------------------------------------------------------------- /snf/layers/distributions/gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/distributions/gaussian.py -------------------------------------------------------------------------------- /snf/layers/distributions/uniform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/distributions/uniform.py -------------------------------------------------------------------------------- /snf/layers/emerging/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /snf/layers/emerging/emerging_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/emerging/emerging_module.py -------------------------------------------------------------------------------- /snf/layers/emerging/inverse_op_cython.cpython-37m-x86_64-linux-gnu.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/emerging/inverse_op_cython.cpython-37m-x86_64-linux-gnu.so -------------------------------------------------------------------------------- /snf/layers/emerging/inverse_op_cython.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/emerging/inverse_op_cython.pyx -------------------------------------------------------------------------------- /snf/layers/emerging/inverse_op_cython.pyxbld: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/emerging/inverse_op_cython.pyxbld -------------------------------------------------------------------------------- /snf/layers/emerging/inverse_triang_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/emerging/inverse_triang_conv.py -------------------------------------------------------------------------------- /snf/layers/emerging/masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/emerging/masks.py -------------------------------------------------------------------------------- /snf/layers/flowlayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/flowlayer.py -------------------------------------------------------------------------------- /snf/layers/flowsequential.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/flowsequential.py -------------------------------------------------------------------------------- /snf/layers/normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/normalize.py -------------------------------------------------------------------------------- /snf/layers/selfnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/selfnorm.py -------------------------------------------------------------------------------- /snf/layers/splines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /snf/layers/splines/rational_quadratic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/splines/rational_quadratic.py -------------------------------------------------------------------------------- /snf/layers/splitprior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/splitprior.py -------------------------------------------------------------------------------- /snf/layers/squeeze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/squeeze.py -------------------------------------------------------------------------------- /snf/layers/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/layers/transforms.py -------------------------------------------------------------------------------- /snf/train/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /snf/train/datatransforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/train/datatransforms.py -------------------------------------------------------------------------------- /snf/train/experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/train/experiment.py -------------------------------------------------------------------------------- /snf/train/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/train/losses.py -------------------------------------------------------------------------------- /snf/train/statsrecorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/train/statsrecorder.py -------------------------------------------------------------------------------- /snf/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /snf/utils/convbackward/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/utils/convbackward/__init__.py -------------------------------------------------------------------------------- /snf/utils/convbackward/conv2d_backward.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/utils/convbackward/conv2d_backward.cpp -------------------------------------------------------------------------------- /snf/utils/convbackward/torch_cpp_extensions/.ninja_deps: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/utils/convbackward/torch_cpp_extensions/.ninja_deps -------------------------------------------------------------------------------- /snf/utils/convbackward/torch_cpp_extensions/.ninja_log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/utils/convbackward/torch_cpp_extensions/.ninja_log -------------------------------------------------------------------------------- /snf/utils/convbackward/torch_cpp_extensions/.placeholder: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /snf/utils/convbackward/torch_cpp_extensions/build.ninja: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/utils/convbackward/torch_cpp_extensions/build.ninja -------------------------------------------------------------------------------- /snf/utils/convbackward/torch_cpp_extensions/conv2d_backward.o: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/utils/convbackward/torch_cpp_extensions/conv2d_backward.o -------------------------------------------------------------------------------- /snf/utils/convbackward/torch_cpp_extensions/conv2d_backward.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/utils/convbackward/torch_cpp_extensions/conv2d_backward.so -------------------------------------------------------------------------------- /snf/utils/toeplitz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/snf/utils/toeplitz.py -------------------------------------------------------------------------------- /tests/snf/test_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akandykeller/SelfNormalizingFlows/HEAD/tests/snf/test_layers.py --------------------------------------------------------------------------------