numpy>=1.25
jax>=0.5.0
flax>=0.10.0
dotmap
matplotlib

[dev]
pytest
pytest-cov
ruff
mypy

[examples]
torch
torchvision
brax
gymnax<2,>=1.0.0
mediapy
ipywidgets
ipykernel
