.coveragerc
.gitignore
.gitlab-ci.yml
.readthedocs.yml
AUTHORS.rst
CHANGELOG.rst
CONTRIBUTING.rst
LICENSE.txt
README.rst
pyproject.toml
setup.cfg
tox.ini
.github/workflows/python-publish.yml
docs/Makefile
docs/authors.rst
docs/changelog.rst
docs/conf.py
docs/contributing.rst
docs/index.rst
docs/license.rst
docs/readme.rst
docs/requirements.txt
docs/sbi.rst
docs/_static/.gitignore
examples/Check_MSD.ipynb
examples/Example Trajectories.ipynb
examples/MMD.ipynb
examples/Train.ipynb
examples/train_lightning.py
src/gratin/__init__.py
src/gratin/standard.py
src/gratin.egg-info/PKG-INFO
src/gratin.egg-info/SOURCES.txt
src/gratin.egg-info/dependency_links.txt
src/gratin.egg-info/not-zip-safe
src/gratin.egg-info/requires.txt
src/gratin.egg-info/top_level.txt
src/gratin/data/__init__.py
src/gratin/data/data.py
src/gratin/data/data_tools.py
src/gratin/data/datamodule.py
src/gratin/data/dataset.py
src/gratin/data/rep_dataset.py
src/gratin/data/skeleton_data.py
src/gratin/layers/InvNet.py
src/gratin/layers/__init__.py
src/gratin/layers/diverse.py
src/gratin/layers/encoders.py
src/gratin/layers/fBMGenerator.py
src/gratin/layers/features_init.py
src/gratin/layers/minimal_conv.py
src/gratin/layers/mmd_pytorch.py
src/gratin/models/BFlowDiffusion.py
src/gratin/models/BFlow_old.py
src/gratin/models/MMD_net.py
src/gratin/models/SelfSup.py
src/gratin/models/__init__.py
src/gratin/models/main_net.py
src/gratin/models/main_net_versatile.py
src/gratin/models/utils.py
src/gratin/models/__pycache__/BFlow.cpython-38.pyc.140382539799328
src/gratin/simulation/__init__.py
src/gratin/simulation/diffusion_models.py
src/gratin/simulation/traj_tools.py
src/gratin/simulation/utils_andi.py
src/gratin/training/__init__.py
src/gratin/training/callbacks.py
src/gratin/training/network_tools.py
src/gratin/training/utils.py
tests/conftest.py