typing-extensions

[benchmark]
jax[cpu]<0.4.0a0,>=0.3
torch<1.13.0a0,>=1.12
torchvision
termcolor

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

[test]
pytest
pytest-cov
pytest-xdist
