numpy>=1.21.0
psutil>=5.8.0

[gpu]
torch>=1.10.0

[viz]
matplotlib
scikit-learn
