torch
einops

[doc]
mkdocs==1.4.3
mkdocs-material==9.1.17
mkdocstrings[python]==0.22.0

[lint]
black
isort
flake8
mccabe
pycodestyle
pyflakes
pep8-naming
flake8-bugbear
flake8-comprehensions
flake8-tidy-imports
darglint
pydocstyle

[test]
backpack-for-pytorch
unfoldNd
jax
jaxlib
coveralls
pytest
pytest-cov
pytest-optional-tests
