numpy
scipy
scikit-learn
matplotlib

[test]
pytest
torch
