torch
numpy
torchvision
networkx
