chex>=0.1.90
jax>=0.8.0

[dev]
build>=1.2.2
flax>=0.10.7
pytest>=8.4
rich>=14.1.0
ruff>=0.12.5
twine>=6.1.0
ty>=0.0.70
tyro>=1.0.6
