.gitignore
Makefile
README.md
requirements-gpu.sh
requirements.txt
setup.py
jais/__init__.py
jais/main.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/jais_settings.json
jais/configs/logs.conf
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/utils/__init__.py
jais/utils/fileloader.py
jais/utils/log.py
jais/utils/utils.py
jais/visualize/data.py
jais/visualize/model.py
logs/training_log@1638551476.125768.csv
logs/training_log@1638551522.1803737.csv
notebooks/.gitkeep
notebooks/scratch.ipynb
tests/conftest.py
tests/test_base.py