einops
torch>=1.8

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