e3nn
torch-cluster
