torch
matplotlib
torchvision
pandas
tqdm
wandb
tensorly
tensorly-torch
tntorch
transformers