numpy
pandas
matplotlib
torch
torchvision
lightning
gradio
grad-cam
torchinfo
torch_lr_finder
pydantic
tqdm
albumentations