numpy

[dev]
pytest
torch

[torch]
torch
