transformers einops --extra-index-url https://download.pytorch.org/whl/cu113 torch shapely timm scipy tokenizers matplotlib