torch
torch_geometric
numpy
matplotlib
networkx
einops
