joblib
numpy
pandas
scikit-learn
scipy

[jax]
jax
jaxlib
dm-haiku
optax

[tensorflow]
tensorflow==2.5.*
tensorflow_probability==0.13.*
tensorflow_addons==0.13.*

[torch]
torch
torchvision
dgl
dgllife
