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

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

[viz]
graphviz
matplotlib
