torch
torchvision
torchmetrics
torch-lr-finder
grad-cam
pillow
numpy
pytorch-lightning