matplotlib
numpy
pandas
seaborn
sklearn
torch>=1.2.0
