numpy
scipy
torch
torchvision
gpustats