torch
torch_npu
einops
