numpy
matplotlib
torch

[test]
pytest>=6.1.2
