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
types-setuptools
flake8
flake8-bugbear
doc8<1.0.0a0
pydocstyle
pyenchant
cpplint
pre-commit

[test]
pytest
pytest-cov
pytest-xdist
jax[cpu]>=0.3
jaxopt
optax
