joblib
numpy>=1.21
pandas
scikit-learn
scipy<1.9
rdkit-pypi

[jax]
jax
jaxlib
dm-haiku
optax

[tensorflow]
tensorflow
tensorflow_probability
tensorflow_addons

[torch]
torch
torchvision
pytorch-lightning
dgl
dgllife
