nn-template-core==0.2.*
pytorch-lightning==1.7.*
torchmetrics==0.10.*
hydra-core==1.2.*
wandb
streamlit
lightning-bolts
rich
dvc
python-dotenv
matplotlib
stqdm
scipy
backports.strenum
pygmtools
einops
torchviz
scikit-learn
seaborn

[dev]
black
flake8
isort
pre-commit
bandit
pytest
pytest-cov
mkdocs
mkdocs-material
mike

[docs]
mkdocs
mkdocs-material
mike

[test]
pytest
pytest-cov
