torch
numpy
torchvision
neptune
torchviz
