numpy
pandas
scikit-learn
datasets
torch
tqdm