torch
torchvision
timm
einops
rasterio