opt-einsum<3.0.0,>=2.3.2
torch<2.0.0,>=1.3.0
torchvision<1.0.0,>=0.3.0
