flytekit<2.0.0,>=1.1.0b0
torch>=1.11.0
