.gitignore
Makefile
README.md
requirements-examples.txt
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/__init__.py
jais/models/example.py
jais/train/__init__.py
jais/train/bs_finder.py
jais/train/log.py
jais/train/losses.py
jais/train/lr_finder.py
jais/train/lr_scheduler.py
jais/train/mixup.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/__init__.py
jais/visualize/data.py
jais/visualize/model.py
logs/cifar10@1638554176.9751344.csv
logs/cifar10@1638555616.6024082.csv
logs/cifar10@1638555918.267136.csv
logs/cifar10@1638556838.9256215.csv
logs/cifar10@1638557654.6198704.csv
logs/cifar10@1638558115.318564.csv
notebooks/.gitkeep
notebooks/scratch.ipynb
tests/conftest.py
tests/test_base.py