torch
numpy
matplotlib
scipy

[test]
pytest
