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

[jaxcpu]
jax>=0.9.0
