numpy
torch
math
