torch>=1.13
optree>=0.4.1
numpy
graphviz
typing-extensions>=4.0.0

[lint]
isort
black[jupyter]>=22.6.0
pylint[spelling]>=2.15.0
mypy>=0.990
flake8
flake8-bugbear
flake8-comprehensions
flake8-docstrings
flake8-pyi
flake8-simplify
ruff
doc8
pydocstyle[toml]
pyenchant
cpplint
pre-commit

[test]
pytest
pytest-cov
pytest-xdist

[test:platform_system != "Windows"]
jax[cpu]>=0.3
jaxopt
optax
