flytekit<2.0.0,>=1.3.0b2
torch>=1.11.0
