torch
torchvision
scikit-learn
pandas
tqdm
python-dotenv
ml_collections
# plots
SciencePlots
seaborn
matplotlib