typing-extensions>=4.0.0

[benchmark]
jax[cpu]<0.5.0a0,>=0.4
torch<1.14.0a0,>=1.13
torchvision
dm-tree<0.2.0a0,>=0.1
pandas
tabulate
termcolor

[lint]
isort>=5.11.0
black>=22.6.0
pylint[spelling]>=2.15.0
mypy>=0.990
flake8
flake8-bugbear
doc8<1.0.0a0
pydocstyle
pyenchant
xdoctest
cpplint
pre-commit

[test]
pytest
pytest-cov
pytest-xdist
