torch>=1.8
einops

[test]
black
mypy
pre-commit
pytest
pytest-cov
ruff
