jax>=0.4.6
chex>=0.1.7
jaxtyping>=0.2.14

[dev]
pytest>=7.1.2
pytest-cov>=4.0.0
twine>=4.0.2
wheel>=0.37.1
tox>=4.4.11
flake8>=6.0.0
mypy>=1.2.0
dm_env>=1.5
