torch
einops
PyWavelets
pytorch-wavelets
diffusers
