numpy
torch
torch_geometric
