accelerate
beartype
einops>=0.4
ema-pytorch
open-clip-torch<3.0.0,>=2.0.0
torch>=1.6
