numpy
scipy
matplotlib

[test]
pytest
torch

[torch]
torch
