typing-extensions>=4.5.0

[benchmark]
jax[cpu]<0.5.0a0,>=0.4.6
torch<2.6.0a0,>=2.0
torchvision
dm-tree<0.2.0a0,>=0.1
pandas
tabulate
termcolor

[docs]
sphinx
sphinx-autoapi
sphinx-autobuild
sphinx-copybutton
sphinx-rtd-theme
sphinxcontrib-bibtex
sphinx-autodoc-typehints
docutils
jax[cpu]
numpy
torch

[jax]
jax

[lint]
black
pylint[spelling]
mypy
flake8
flake8-bugbear
flake8-comprehensions
flake8-docstrings
flake8-pyi
flake8-simplify
ruff
doc8
pydocstyle
pyenchant
xdoctest
cpplint
pre-commit

[numpy]
numpy

[test]
pytest
pytest-cov
pytest-xdist

[torch]
torch
