torch
einops
PyWavelets
timm
torchvision
pillow
