chex
jax
numpy
treescope
xarray

[complete]
jax-datetime

[tests]
absl-py
pytest
