pywt
torch
numpy
