torch>=2.1
einops
onnxscript
onnx
