numpy
torch>=1.10.0
sympy
