torch
einops
torchvision
opencv-python
kornia
albumentations
loguru
tqdm
matplotlib
h5py
wandb
timm
xformers # Optional, used for memefficient attention