numpy
pandas
torch
scikit-learn>=0.22
plotly
pynvml
nni
gitpython
