hydra-core>=1.3
torch
numpy
scikit-learn
tqdm
omegaconf

[docs]
mkdocs<2,>=1.6
mkdocs-material>=9
mkdocstrings[python]>=0.26

[test]
pytest>=8
pytest-cov>=5
pandas
torchsonn[viz]

[viz]
graphviz
matplotlib
