numpy
typing
torch

[test]
pytest
