torch
torchvision
xarray
numpy
