flytekit<1.2.0,>=1.0.0b0
torch>=1.11.0
