
[dev]
build
flake8
isort
mypy>=1.0.0
pytest
regex
toml
twine
types-regex
types-setuptools
yapf

[dev:python_version >= "3.11"]
jax>=0.9.0

[jaxcpu]

[jaxcpu:python_version >= "3.11"]
jax>=0.9.0
