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

[:python_version < "3.9"]
networkx<3.2
