torch
torchvision
einops
numpy
pillow
tifffile

[train]
pytorch-lightning
torchmetrics
