einops>=0.3
torch>=1.6
torch-scatter
torch-sparse
torch-cluster
torch-spline-conv
torch-geometric
