numpy
pytest
torch
