jax
jaxlib
numpy
pint
scipy
scikit-learn
tree-math
dask
fsspec
pandas
xarray
xarray-tensorstore

[tests]
absl-py
chex
pytest
