flytekit<2.0.0,>=1.3.0b2
torch>=1.11.0
onnx-pytorch
onnxscript
numpy<2.0.0

[:python_version < "3.9"]
networkx<3.2
