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

[dev]
pytest>=8.3.2
pytest-cov>=5.0.0
ruff>=0.6.4
mypy>=1.11.2

[examples]
torch
torchvision
brax
gymnax
ipykernel
