numpy
typing
torch