torch>=2.1
numpy

[parity]
zennit

[test]
pytest
pytest-randomly
