numpy>=1.25
jax<0.7,>=0.5.0
flax<0.11,>=0.10.0
orbax-checkpoint<0.11
dotmap
matplotlib

[dev]
pytest
pytest-cov
ruff
mypy

[examples]
torch
torchvision
brax
gymnax>=0.0.9
mediapy
ipywidgets
ipykernel
