├── .github └── workflows │ ├── lint.yml │ └── test.yml ├── .gitignore ├── .pre-commit-config.yaml ├── assets ├── maf │ └── moons │ │ ├── point-flow.pdf │ │ ├── point-flow.png │ │ ├── z2x+x2z.pdf │ │ └── z2x+x2z.png ├── mnf │ └── mnist │ │ ├── rot-9-lenet-s.png │ │ ├── rot-9-lenet.pdf │ │ ├── rot-9-lenet.png │ │ ├── rot-9-mnf-lenet-s.png │ │ ├── rot-9-mnf-lenet.pdf │ │ └── rot-9-mnf-lenet.png ├── nsf_ar │ └── moons │ │ ├── point-flow.pdf │ │ ├── point-flow.png │ │ ├── z2x+x2z.pdf │ │ └── z2x+x2z.png ├── nsf_cl │ └── moons │ │ ├── point-flow.pdf │ │ ├── point-flow.png │ │ ├── z2x+x2z.pdf │ │ └── z2x+x2z.png └── rnvp │ └── moons │ ├── point-flow.pdf │ ├── point-flow.png │ ├── z2x+x2z.pdf │ └── z2x+x2z.png ├── data └── siggraph.pkl ├── examples ├── half_moons.ipynb └── mnf_mnist.ipynb ├── license ├── pyproject.toml ├── readme.md ├── tests ├── test_flows.py └── test_mnf_mnist.py └── torch_mnf ├── __init__.py ├── data.py ├── flows ├── __init__.py ├── affine_constant_flow.py ├── affine_half_flow.py ├── core.py ├── glow.py ├── maf.py ├── rnvp.py └── spline_flow.py ├── layers ├── __init__.py ├── made.py ├── mnf_conv.py └── mnf_linear.py ├── models ├── __init__.py ├── lenet.py ├── mlp.py ├── mnf_feed_forward.py └── mnf_lenet.py └── utils.py /.github/workflows/lint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/.github/workflows/lint.yml -------------------------------------------------------------------------------- /.github/workflows/test.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/.github/workflows/test.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | data/MNIST 3 | runs 4 | *.egg-info 5 | *logs 6 | build/ 7 | -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /assets/maf/moons/point-flow.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/maf/moons/point-flow.pdf -------------------------------------------------------------------------------- /assets/maf/moons/point-flow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/maf/moons/point-flow.png -------------------------------------------------------------------------------- /assets/maf/moons/z2x+x2z.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/maf/moons/z2x+x2z.pdf -------------------------------------------------------------------------------- /assets/maf/moons/z2x+x2z.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/maf/moons/z2x+x2z.png -------------------------------------------------------------------------------- /assets/mnf/mnist/rot-9-lenet-s.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/mnf/mnist/rot-9-lenet-s.png -------------------------------------------------------------------------------- /assets/mnf/mnist/rot-9-lenet.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/mnf/mnist/rot-9-lenet.pdf -------------------------------------------------------------------------------- /assets/mnf/mnist/rot-9-lenet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/mnf/mnist/rot-9-lenet.png -------------------------------------------------------------------------------- /assets/mnf/mnist/rot-9-mnf-lenet-s.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/mnf/mnist/rot-9-mnf-lenet-s.png -------------------------------------------------------------------------------- /assets/mnf/mnist/rot-9-mnf-lenet.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/mnf/mnist/rot-9-mnf-lenet.pdf -------------------------------------------------------------------------------- /assets/mnf/mnist/rot-9-mnf-lenet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/mnf/mnist/rot-9-mnf-lenet.png -------------------------------------------------------------------------------- /assets/nsf_ar/moons/point-flow.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/nsf_ar/moons/point-flow.pdf -------------------------------------------------------------------------------- /assets/nsf_ar/moons/point-flow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/nsf_ar/moons/point-flow.png -------------------------------------------------------------------------------- /assets/nsf_ar/moons/z2x+x2z.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/nsf_ar/moons/z2x+x2z.pdf -------------------------------------------------------------------------------- /assets/nsf_ar/moons/z2x+x2z.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/nsf_ar/moons/z2x+x2z.png -------------------------------------------------------------------------------- /assets/nsf_cl/moons/point-flow.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/nsf_cl/moons/point-flow.pdf -------------------------------------------------------------------------------- /assets/nsf_cl/moons/point-flow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/nsf_cl/moons/point-flow.png -------------------------------------------------------------------------------- /assets/nsf_cl/moons/z2x+x2z.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/nsf_cl/moons/z2x+x2z.pdf -------------------------------------------------------------------------------- /assets/nsf_cl/moons/z2x+x2z.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/nsf_cl/moons/z2x+x2z.png -------------------------------------------------------------------------------- /assets/rnvp/moons/point-flow.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/rnvp/moons/point-flow.pdf -------------------------------------------------------------------------------- /assets/rnvp/moons/point-flow.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/rnvp/moons/point-flow.png -------------------------------------------------------------------------------- /assets/rnvp/moons/z2x+x2z.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/rnvp/moons/z2x+x2z.pdf -------------------------------------------------------------------------------- /assets/rnvp/moons/z2x+x2z.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/assets/rnvp/moons/z2x+x2z.png -------------------------------------------------------------------------------- /data/siggraph.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/data/siggraph.pkl -------------------------------------------------------------------------------- /examples/half_moons.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/examples/half_moons.ipynb -------------------------------------------------------------------------------- /examples/mnf_mnist.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/examples/mnf_mnist.ipynb -------------------------------------------------------------------------------- /license: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/license -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/pyproject.toml -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/readme.md -------------------------------------------------------------------------------- /tests/test_flows.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/tests/test_flows.py -------------------------------------------------------------------------------- /tests/test_mnf_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/tests/test_mnf_mnist.py -------------------------------------------------------------------------------- /torch_mnf/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /torch_mnf/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/data.py -------------------------------------------------------------------------------- /torch_mnf/flows/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/flows/__init__.py -------------------------------------------------------------------------------- /torch_mnf/flows/affine_constant_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/flows/affine_constant_flow.py -------------------------------------------------------------------------------- /torch_mnf/flows/affine_half_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/flows/affine_half_flow.py -------------------------------------------------------------------------------- /torch_mnf/flows/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/flows/core.py -------------------------------------------------------------------------------- /torch_mnf/flows/glow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/flows/glow.py -------------------------------------------------------------------------------- /torch_mnf/flows/maf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/flows/maf.py -------------------------------------------------------------------------------- /torch_mnf/flows/rnvp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/flows/rnvp.py -------------------------------------------------------------------------------- /torch_mnf/flows/spline_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/flows/spline_flow.py -------------------------------------------------------------------------------- /torch_mnf/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/layers/__init__.py -------------------------------------------------------------------------------- /torch_mnf/layers/made.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/layers/made.py -------------------------------------------------------------------------------- /torch_mnf/layers/mnf_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/layers/mnf_conv.py -------------------------------------------------------------------------------- /torch_mnf/layers/mnf_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/layers/mnf_linear.py -------------------------------------------------------------------------------- /torch_mnf/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/models/__init__.py -------------------------------------------------------------------------------- /torch_mnf/models/lenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/models/lenet.py -------------------------------------------------------------------------------- /torch_mnf/models/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/models/mlp.py -------------------------------------------------------------------------------- /torch_mnf/models/mnf_feed_forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/models/mnf_feed_forward.py -------------------------------------------------------------------------------- /torch_mnf/models/mnf_lenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/models/mnf_lenet.py -------------------------------------------------------------------------------- /torch_mnf/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/janosh/torch-mnf/HEAD/torch_mnf/utils.py --------------------------------------------------------------------------------