numpy>=1.16
scipy
scikit-learn
torch
