.coveragerc
.gitignore
.pre-commit-config.yaml
LICENCE.txt
MANIFEST.in
Makefile
README.md
pyproject.toml
requirements.txt
schedule.py
setup.cfg
setup.py
tables.ipynb
todo.tasks
train.py
.github/workflows/build_book.yml
.github/workflows/ci.yml
.github/workflows/publish.yml
.github/workflows/style.yml
docs/_config.yml
docs/_toc.yml
docs/advanced_usage.md
docs/architectures.rst
docs/basic_usage.md
docs/build_your_own_model.md
docs/coders.rst
docs/intro.md
docs/logo.png
docs/references.bib
experiment/__init__.py
experiment/plot.py
experiment/util.py
experiment/data/__init__.py
experiment/data/antarctica.py
experiment/data/eeg.py
experiment/data/gp.py
experiment/data/predprey.py
experiment/data/temperature.py
experiment/data/util.py
neuralprocesses/__init__.py
neuralprocesses/_version.py
neuralprocesses/aggregate.py
neuralprocesses/augment.py
neuralprocesses/chain.py
neuralprocesses/coding.py
neuralprocesses/datadims.py
neuralprocesses/disc.py
neuralprocesses/likelihood.py
neuralprocesses/mask.py
neuralprocesses/materialise.py
neuralprocesses/numdata.py
neuralprocesses/parallel.py
neuralprocesses/util.py
neuralprocesses.egg-info/PKG-INFO
neuralprocesses.egg-info/SOURCES.txt
neuralprocesses.egg-info/dependency_links.txt
neuralprocesses.egg-info/requires.txt
neuralprocesses.egg-info/top_level.txt
neuralprocesses/architectures/__init__.py
neuralprocesses/architectures/agnp.py
neuralprocesses/architectures/climate.py
neuralprocesses/architectures/convgnp.py
neuralprocesses/architectures/fullconvgnp.py
neuralprocesses/architectures/gnp.py
neuralprocesses/architectures/util.py
neuralprocesses/coders/__init__.py
neuralprocesses/coders/aggregate.py
neuralprocesses/coders/attention.py
neuralprocesses/coders/augment.py
neuralprocesses/coders/copy.py
neuralprocesses/coders/deepset.py
neuralprocesses/coders/densecov.py
neuralprocesses/coders/functional.py
neuralprocesses/coders/fuse.py
neuralprocesses/coders/inputs.py
neuralprocesses/coders/mapdiag.py
neuralprocesses/coders/nn.py
neuralprocesses/coders/shaping.py
neuralprocesses/coders/setconv/__init__.py
neuralprocesses/coders/setconv/density.py
neuralprocesses/coders/setconv/identity.py
neuralprocesses/coders/setconv/setconv.py
neuralprocesses/data/__init__.py
neuralprocesses/data/antarctica.py
neuralprocesses/data/batch.py
neuralprocesses/data/data.py
neuralprocesses/data/eeg.py
neuralprocesses/data/gp.py
neuralprocesses/data/mixgp.py
neuralprocesses/data/mixture.py
neuralprocesses/data/predefined.py
neuralprocesses/data/predprey.py
neuralprocesses/data/sawtooth.py
neuralprocesses/data/temperature.py
neuralprocesses/data/util.py
neuralprocesses/dist/__init__.py
neuralprocesses/dist/dirac.py
neuralprocesses/dist/dist.py
neuralprocesses/dist/geom.py
neuralprocesses/dist/normal.py
neuralprocesses/dist/transformed.py
neuralprocesses/dist/uniform.py
neuralprocesses/model/__init__.py
neuralprocesses/model/ar.py
neuralprocesses/model/elbo.py
neuralprocesses/model/loglik.py
neuralprocesses/model/model.py
neuralprocesses/model/predict.py
neuralprocesses/model/util.py
neuralprocesses/tensorflow/__init__.py
neuralprocesses/tensorflow/nn.py
neuralprocesses/torch/__init__.py
neuralprocesses/torch/nn.py
scripts/predprey.py
scripts/predprey_visualise.py
scripts/sawtooth_sample_ar.py
scripts/synthetic_extra.py
scripts/synthetic_parse.py
scripts/temperature_mae.py
scripts/temperature_summarise_folds.py
scripts/temperature_visualise.py
tests/__init__.py
tests/test_architectures.py
tests/test_augment.py
tests/test_chain.py
tests/test_data.py
tests/test_discretisation.py
tests/test_distribution.py
tests/test_mask.py
tests/test_model.py
tests/test_unet.py
tests/test_util.py
tests/util.py
tests/coders/__init__.py
tests/coders/test_shaping.py
tests/dists/__init__.py
tests/dists/test_normal.py
tests/gnp/__init__.py
tests/gnp/autoencoding.py
tests/gnp/gnp.py
tests/gnp/util.py