Concerns about implement WNN (Wavelet Neural Network) in PyTorch

Hi everyone,
I searched the docs but didn’t find anything about WNN (Wavelet Neural Network) in PyTorch, does PyTorch include it? Or if I implement it manually, do you have some suggestions about how to implement it in PyTorch efficiently? Or are there libraries that can realize WNN? Thank you.