torch
einops
PyWavelets
torchvision
