rdkit
torch
numpy
torch-geometric
torch-scatter
torch-sparse
