torch>=2.0
numpy
scipy
scikit-learn
tqdm
psutil

[viz]
matplotlib
seaborn
