numpy
scipy
torch

[dev]
pytest

[plotting]
matplotlib
