einops
torch
torchvision
pywavelets
numpy
