torch
torchvision
captum
matplotlib