torch>=2.1
einops
onnxscript==0.3.1
onnx
