torch
timm
segmentation_models_pytorch
