typing-extensions>=4.0.0

[benchmark]
jax[cpu]<0.4.0a0,>=0.3
torch<1.14.0a0,>=1.13
torchvision
dm-tree<0.2.0a0,>=0.1.0
pandas
tabulate
termcolor

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

[test]
pytest
pytest-cov
pytest-xdist
