numpy
torch
torchvision
timm
tensorboard
sklearn
pillow
