├── .gitattributes ├── .github ├── ISSUE_TEMPLATE │ ├── bug-report.md │ ├── documentation.md │ └── feature-request.md └── workflows │ ├── ci.yml │ └── dummy.yml ├── .gitignore ├── CITATION.bib ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── docs ├── api.rst ├── conf.py ├── docutils.conf ├── images │ ├── banner.svg │ └── banner_dark.svg ├── index.rst ├── static │ ├── custom.css │ └── logo.svg ├── templates │ └── autosummary │ │ └── module.rst ├── tutorials.rst └── tutorials │ ├── basics.ipynb │ ├── bayesian_flows.ipynb │ ├── forward_kl.ipynb │ ├── reverse_kl.ipynb │ └── vae.ipynb ├── pre-commit.yml ├── pyproject.toml ├── readthedocs.yml ├── tests ├── conftest.py ├── test_bayesian.py ├── test_distributions.py ├── test_flows.py ├── test_mixtures.py ├── test_nn.py ├── test_transforms.py └── test_utils.py └── zuko ├── __init__.py ├── bayesian.py ├── conftest.py ├── distributions.py ├── flows ├── __init__.py ├── autoregressive.py ├── continuous.py ├── core.py ├── coupling.py ├── gaussianization.py ├── mixture.py ├── neural.py ├── polynomial.py └── spline.py ├── lazy.py ├── mixtures.py ├── nn.py ├── transforms.py └── utils.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/.gitattributes -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug-report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/.github/ISSUE_TEMPLATE/bug-report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/documentation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/.github/ISSUE_TEMPLATE/documentation.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature-request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/.github/ISSUE_TEMPLATE/feature-request.md -------------------------------------------------------------------------------- /.github/workflows/ci.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/.github/workflows/ci.yml -------------------------------------------------------------------------------- /.github/workflows/dummy.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/.github/workflows/dummy.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/.gitignore -------------------------------------------------------------------------------- /CITATION.bib: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/CITATION.bib -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/README.md -------------------------------------------------------------------------------- /docs/api.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/docs/api.rst -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/docutils.conf: -------------------------------------------------------------------------------- 1 | [restructuredtext parser] 2 | syntax_highlight = short 3 | -------------------------------------------------------------------------------- /docs/images/banner.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/docs/images/banner.svg -------------------------------------------------------------------------------- /docs/images/banner_dark.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/docs/images/banner_dark.svg -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/static/custom.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/docs/static/custom.css -------------------------------------------------------------------------------- /docs/static/logo.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/docs/static/logo.svg -------------------------------------------------------------------------------- /docs/templates/autosummary/module.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/docs/templates/autosummary/module.rst -------------------------------------------------------------------------------- /docs/tutorials.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/docs/tutorials.rst -------------------------------------------------------------------------------- /docs/tutorials/basics.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/docs/tutorials/basics.ipynb -------------------------------------------------------------------------------- /docs/tutorials/bayesian_flows.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/docs/tutorials/bayesian_flows.ipynb -------------------------------------------------------------------------------- /docs/tutorials/forward_kl.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/docs/tutorials/forward_kl.ipynb -------------------------------------------------------------------------------- /docs/tutorials/reverse_kl.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/docs/tutorials/reverse_kl.ipynb -------------------------------------------------------------------------------- /docs/tutorials/vae.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/docs/tutorials/vae.ipynb -------------------------------------------------------------------------------- /pre-commit.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/pre-commit.yml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/pyproject.toml -------------------------------------------------------------------------------- /readthedocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/readthedocs.yml -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/test_bayesian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/tests/test_bayesian.py -------------------------------------------------------------------------------- /tests/test_distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/tests/test_distributions.py -------------------------------------------------------------------------------- /tests/test_flows.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/tests/test_flows.py -------------------------------------------------------------------------------- /tests/test_mixtures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/tests/test_mixtures.py -------------------------------------------------------------------------------- /tests/test_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/tests/test_nn.py -------------------------------------------------------------------------------- /tests/test_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/tests/test_transforms.py -------------------------------------------------------------------------------- /tests/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/tests/test_utils.py -------------------------------------------------------------------------------- /zuko/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/__init__.py -------------------------------------------------------------------------------- /zuko/bayesian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/bayesian.py -------------------------------------------------------------------------------- /zuko/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/conftest.py -------------------------------------------------------------------------------- /zuko/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/distributions.py -------------------------------------------------------------------------------- /zuko/flows/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/flows/__init__.py -------------------------------------------------------------------------------- /zuko/flows/autoregressive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/flows/autoregressive.py -------------------------------------------------------------------------------- /zuko/flows/continuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/flows/continuous.py -------------------------------------------------------------------------------- /zuko/flows/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/flows/core.py -------------------------------------------------------------------------------- /zuko/flows/coupling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/flows/coupling.py -------------------------------------------------------------------------------- /zuko/flows/gaussianization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/flows/gaussianization.py -------------------------------------------------------------------------------- /zuko/flows/mixture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/flows/mixture.py -------------------------------------------------------------------------------- /zuko/flows/neural.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/flows/neural.py -------------------------------------------------------------------------------- /zuko/flows/polynomial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/flows/polynomial.py -------------------------------------------------------------------------------- /zuko/flows/spline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/flows/spline.py -------------------------------------------------------------------------------- /zuko/lazy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/lazy.py -------------------------------------------------------------------------------- /zuko/mixtures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/mixtures.py -------------------------------------------------------------------------------- /zuko/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/nn.py -------------------------------------------------------------------------------- /zuko/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/transforms.py -------------------------------------------------------------------------------- /zuko/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/probabilists/zuko/HEAD/zuko/utils.py --------------------------------------------------------------------------------