torch
torch_geometric
rdkit