chex>=0.1.89
flax>=0.10.5
jax<0.6.0,>=0.4.25
jaxlib>=0.5.3
numpy>=1.26.4

[all]
stoa-env[brax]
stoa-env[gymnax]
stoa-env[jumanji]
stoa-env[kinetix]
stoa-env[navix]
stoa-env[pgx]
stoa-env[xminigrid]

[brax]
brax>=0.9.0

[dev]
black
coverage
flake8
importlib-metadata
isort
livereload
mkdocs
mkdocs-git-revision-date-plugin
mkdocs-include-markdown-plugin
mkdocs-material
mkdocs-mermaid2-plugin
mkdocstrings
mknotebooks
mypy
nbmake
pre-commit
promise
pymdown-extensions
pytest
pytest-cov
pytest-mock
pytest-parallel
pytest-xdist
pytype
testfixtures

[gymnax]
gymnax>=0.0.6
gymnasium>=1.1.1

[jumanji]
jumanji==1.0.0

[kinetix]
kinetix-env

[navix]
navix>=0.7.0

[pgx]
pgx>=2.6.0

[xminigrid]
xminigrid
