├── .github └── workflows │ └── pytest_and_autopublish.yml ├── .gitignore ├── .pylintrc ├── .readthedocs.yaml ├── .vscode ├── extensions.json └── settings.json ├── AUTHORS ├── CHANGELOG.md ├── CONTRIBUTING.md ├── LICENSE ├── Makefile ├── README.md ├── coix ├── __init__.py ├── algo.py ├── algo_test.py ├── api.py ├── api_test.py ├── core.py ├── core_test.py ├── loss.py ├── loss_test.py ├── numpyro.py ├── oryx.py ├── oryx_test.py ├── util.py └── util_test.py ├── docs ├── Makefile ├── _static │ ├── anneal.png │ ├── anneal_oryx.png │ ├── bmnist.gif │ ├── dmm.png │ ├── dmm_oryx.png │ ├── gmm.png │ ├── gmm_oryx.png │ ├── tutorial_part1_vae.png │ ├── tutorial_part2_api.png │ └── tutorial_part3_smcs.png ├── _templates │ └── breadcrumbs.html ├── algo.rst ├── api.rst ├── conf.py ├── core.rst ├── index.rst ├── loss.rst ├── requirements.txt └── util.rst ├── examples ├── README.rst ├── anneal.py ├── anneal_oryx.py ├── bmnist.py ├── dmm.py ├── dmm_oryx.py ├── gmm.py └── gmm_oryx.py ├── notebooks ├── figures │ └── smcs_nvi.png ├── tutorial_part1_vae.ipynb ├── tutorial_part2_api.ipynb └── tutorial_part3_smcs.ipynb └── pyproject.toml /.github/workflows/pytest_and_autopublish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/.github/workflows/pytest_and_autopublish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/.gitignore -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/.pylintrc -------------------------------------------------------------------------------- /.readthedocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/.readthedocs.yaml -------------------------------------------------------------------------------- /.vscode/extensions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/.vscode/extensions.json -------------------------------------------------------------------------------- /.vscode/settings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/.vscode/settings.json -------------------------------------------------------------------------------- /AUTHORS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/AUTHORS -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/README.md -------------------------------------------------------------------------------- /coix/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/coix/__init__.py -------------------------------------------------------------------------------- /coix/algo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/coix/algo.py -------------------------------------------------------------------------------- /coix/algo_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/coix/algo_test.py -------------------------------------------------------------------------------- /coix/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/coix/api.py -------------------------------------------------------------------------------- /coix/api_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/coix/api_test.py -------------------------------------------------------------------------------- /coix/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/coix/core.py -------------------------------------------------------------------------------- /coix/core_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/coix/core_test.py -------------------------------------------------------------------------------- /coix/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/coix/loss.py -------------------------------------------------------------------------------- /coix/loss_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/coix/loss_test.py -------------------------------------------------------------------------------- /coix/numpyro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/coix/numpyro.py -------------------------------------------------------------------------------- /coix/oryx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/coix/oryx.py -------------------------------------------------------------------------------- /coix/oryx_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/coix/oryx_test.py -------------------------------------------------------------------------------- /coix/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/coix/util.py -------------------------------------------------------------------------------- /coix/util_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/coix/util_test.py -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/_static/anneal.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/_static/anneal.png -------------------------------------------------------------------------------- /docs/_static/anneal_oryx.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/_static/anneal_oryx.png -------------------------------------------------------------------------------- /docs/_static/bmnist.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/_static/bmnist.gif -------------------------------------------------------------------------------- /docs/_static/dmm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/_static/dmm.png -------------------------------------------------------------------------------- /docs/_static/dmm_oryx.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/_static/dmm_oryx.png -------------------------------------------------------------------------------- /docs/_static/gmm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/_static/gmm.png -------------------------------------------------------------------------------- /docs/_static/gmm_oryx.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/_static/gmm_oryx.png -------------------------------------------------------------------------------- /docs/_static/tutorial_part1_vae.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/_static/tutorial_part1_vae.png -------------------------------------------------------------------------------- /docs/_static/tutorial_part2_api.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/_static/tutorial_part2_api.png -------------------------------------------------------------------------------- /docs/_static/tutorial_part3_smcs.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/_static/tutorial_part3_smcs.png -------------------------------------------------------------------------------- /docs/_templates/breadcrumbs.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/_templates/breadcrumbs.html -------------------------------------------------------------------------------- /docs/algo.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/algo.rst -------------------------------------------------------------------------------- /docs/api.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/api.rst -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/core.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/core.rst -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/loss.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/loss.rst -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/util.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/docs/util.rst -------------------------------------------------------------------------------- /examples/README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/examples/README.rst -------------------------------------------------------------------------------- /examples/anneal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/examples/anneal.py -------------------------------------------------------------------------------- /examples/anneal_oryx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/examples/anneal_oryx.py -------------------------------------------------------------------------------- /examples/bmnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/examples/bmnist.py -------------------------------------------------------------------------------- /examples/dmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/examples/dmm.py -------------------------------------------------------------------------------- /examples/dmm_oryx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/examples/dmm_oryx.py -------------------------------------------------------------------------------- /examples/gmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/examples/gmm.py -------------------------------------------------------------------------------- /examples/gmm_oryx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/examples/gmm_oryx.py -------------------------------------------------------------------------------- /notebooks/figures/smcs_nvi.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/notebooks/figures/smcs_nvi.png -------------------------------------------------------------------------------- /notebooks/tutorial_part1_vae.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/notebooks/tutorial_part1_vae.ipynb -------------------------------------------------------------------------------- /notebooks/tutorial_part2_api.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/notebooks/tutorial_part2_api.ipynb -------------------------------------------------------------------------------- /notebooks/tutorial_part3_smcs.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/notebooks/tutorial_part3_smcs.ipynb -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jax-ml/coix/HEAD/pyproject.toml --------------------------------------------------------------------------------