torch
einops
PyWavelets
pytorch-wavelets
