numpy
scipy

[autograd]
autograd>=1.3

[ci]
pytest-github-actions-annotate-failures

[pytorch]
torch>=1.9.1

[test]
pytest
pytest-cov
coverage
geomstats
nbformat
nbconvert
