CONTRIBUTING.md
LICENSE
MANIFEST.in
README.md
settings.ini
setup.py
jax_relax.egg-info/PKG-INFO
jax_relax.egg-info/SOURCES.txt
jax_relax.egg-info/dependency_links.txt
jax_relax.egg-info/entry_points.txt
jax_relax.egg-info/not-zip-safe
jax_relax.egg-info/requires.txt
jax_relax.egg-info/top_level.txt
relax/__init__.py
relax/_ckpt_manager.py
relax/_modidx.py
relax/evaluate.py
relax/import_essentials.py
relax/logger.py
relax/module.py
relax/plots.py
relax/trainer.py
relax/utils.py
relax/data/__init__.py
relax/data/loader.py
relax/data/module.py
relax/methods/__init__.py
relax/methods/base.py
relax/methods/counternet.py
relax/methods/diverse.py
relax/methods/proto.py
relax/methods/vanilla.py