attrs
chex
dpshdl
equinox
importlib-resources
jax
jaxtyping
optax
orbax-checkpoint
pillow
omegaconf
gitpython
tensorboard
psutil
requests

[all]
black
darglint
mypy
ruff
pytest
types-pillow
types-psutil
types-requests
flax
orbax-export
tensorflow

[dev]
black
darglint
mypy
ruff
pytest
types-pillow
types-psutil
types-requests

[exportable]
flax
orbax-export
tensorflow
