torch>=1.8.1
torchvision
pytorch-lightning==1.8.6
torchmetrics>=0.3.2
lightning-bolts>=0.3.3
hydra-core==1.2.0
hydra-colorlog==1.1.0.dev1
hydra-optuna-sweeper==1.1.0.dev2
wandb>=0.12.21
pre-commit
black
isort
flake8
ipdb>=0.13.9
loguru>=0.6.0
jupyterlab
pyrootutils
rich
pytest
sh
einops
scikit-learn
seaborn
pudb
colorlog
graphviz>=0.20.1
peewee
