Geospatial Deep Learning with TorchGeo
pytorch.org