numpy==1.23.5
vmas
wandb
torch
torch_geometric
torch_scatter
torch_sparse
torch_cluster
torch_spline_conv