PyWavelets
torch
numpy
