numpy
torch
torch-geometric
torch-scatter
torch-sparse
torch-cluster
torch-spline-conv
pyyaml
