pip
packaging
torch
numpy
