.gitignore
LICENSE
MANIFEST.in
Makefile
README.md
requirements-base.txt
requirements-gpu.sh
requirements.txt
setup.py
test.py
docs/Makefile
docs/make.bat
docs/source/conf.py
docs/source/index.rst
docs/source/jais.rst
docs/source/modules.rst
jais/__init__.py
jais/__main__.py
jais/utils.py
jais.egg-info/PKG-INFO
jais.egg-info/SOURCES.txt
jais.egg-info/dependency_links.txt
jais.egg-info/entry_points.txt
jais.egg-info/requires.txt
jais.egg-info/top_level.txt
jais/configs/default.yaml
jais/configs/logs.conf
jais/data/dataset.py
jais/examples/cifar10.py
jais/models/example.py
jais/train/bs_finder.py
jais/train/log.py
jais/train/lr_finder.py
jais/train/lr_scheduler.py
jais/train/trainer.py
jais/train/tune_hparams.py
jais/visualize/data.py
jais/visualize/model.py
logs/.gitkeep
logs/training_logs@1638006825.9317725.csv
notebooks/.gitkeep
notebooks/scratch.ipynb
tests/.gitkeep
tests/conftest.py
tests/test_base.py