torch
torchvision
torchaudio
numpy
matplotlib
captum
tqdm
