torch
einops
PyWavelets
