dinosaur>=1.3.3
dm-haiku
gin-config
jax
jaxlib
numpy
optax
pandas
xarray

[experimental]
coordax
einops
etils
fiddle
flax
grain
jax-datetime
jax-solar
orbax
treescope

[tests]
absl-py
chex
pytest
