numpy

[dev]
pytest
pytest-xdist
torch
matplotlib

[torch]
torch

[vis]
matplotlib
