torch
torchvision
scikit-image
einops