jax
numpy
treescope

[complete]
chex
jax-datetime
xarray

[tests]
absl-py
pytest
chex
