jax
jaxlib
numpy
pint
scipy
scikit-learn
tree-math

[tests]
absl-py
chex
pytest

[xarray-utils]
dask
fsspec
pandas
xarray
xarray-tensorstore
