pytest
tensorflow-cpu
torch
numpy
