joblib
numpy
pandas
scikit-learn
scipy

[jax]
jax==0.2.14
jaxlib==0.1.67
dm-haiku==0.0.4
optax==0.0.8

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

[torch]
torch
torchvision
dgl
dgllife
