torch
torch-geometric
torch-scatter
