transformers
torch
timm
segmentation-models-pytorch
einops
numpy