numpy>=1.17.2
pandas
dask[array]
torch>=1.12
jax>=0.4.8
jaxlib>=0.4.7
