jax
jaxtyping
equinox
optax
dpshdl
chex
importlib-resources
cloudpickle
pillow
omegaconf
gitpython
tensorboard
psutil
requests

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