torch
torchvision
timm
einops
rasterio
git+https://github.com/IBM/terratorch.git