torch>=1.12
spacetransformer-core>=0.2.1

[dev]
pytest

[test]
pytest
