numpy>=1.17.3
pandas
scipy>=1.0
scikit-learn>=0.20
dask[array]

[all]
torch>=1.12
jax>=0.4.8
jaxlib>=0.4.7
tensorflow>=2.16

[jax]
jax>=0.4.8
jaxlib>=0.4.7

[tensorflow]
tensorflow>=2.16

[torch]
torch>=1.12
