torch
timm
einops
segmentation-models-pytorch
