bitorchinfo
matplotlib
numpy
sklearn
torch
torchvision
