torch>=1.9.0 torch-geometric>=2.0.0 rdkit-pypi>=2022.3.0 numpy>=1.21.0