jax
numpy
scipy

[complete]
matplotlib
seaborn
Pillow
xarray
dm-haiku
einops
gin-config
absl-py
pytest
pytest-xdist
scikit-image

[data]
matplotlib
seaborn
Pillow
xarray

[ml]
dm-haiku
einops
gin-config

[tests]
absl-py
pytest
pytest-xdist
scikit-image
