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

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

[wandb]
wandb[media]
