torch
torchvision
numpy
tqdm
matplotlib
entmax
wandb
tabulate
scipy
