torch
torchvision
timm
einops
rasterio
terratorch==0.99.8