numpy
pandas
torch
scikit-learn
plotly
pynvml
