torch==2.4.0
torch_geometric==2.6.1
rdkit
matplotlib