torch==1.12.1 torchvision==0.13.1 timm==0.6.12 Pillow blobfile mypy numpy pytest requests einops deepspeed==0.4.0 scipy