torch>=2.2 torchvision safetensors>=0.4 timm==1.0.8 typed-argument-parser==1.12.0 pytz numpy