tifffile
torch
torchvision
einops
