dinosaur-dycore>=1.1.1
dm-haiku
gin-config
jax
jaxlib
numpy
optax
pandas
tensorflow_probability
xarray

[experimental]
etils
fiddle
flax
orbax
penzai

[tests]
absl-py
chex
pytest
