torch>=1.8.0
opt_einsum
