torch>=2.1.0
torchvision
torchdata
iopath
tensordict

[benchmarks]
pytest-benchmark
tenacity

[docs]
sphinx
sphinx-rtd-theme

[notebooks]
ipython
jupyter
ipython-autotime
imutils
matplotlib
seaborn

[tests]
hypothesis
bandit[toml]
black>=23.10.0
check-manifest
flake8
flake8-bugbear
flake8-docstrings
flake8-formatter_junit_xml
flake8-pyproject
pre-commit
pylint
pylint_junit
pytest
pytest-sugar
pytest-xdist
pytest-benchmark
pytest-cov
pytest-mock
pytest-runner
pytest-github-actions-annotate-failures
shellcheck-py
mypy
autoflake
pydocstyle
pyre-check
pydocstringformatter
pyrefact
