gpustat
matplotlib
numpy
pandas
pytorch_lightning
scikit_learn
seaborn
tensorboard
torch
tqdm
xgboost
