jax
numpy
treescope
typing_extensions

[complete]
chex
jax-datetime
xarray

[tests]
absl-py
pytest
chex
