opt-einsum
torch
numpy
