refactor: implement cwt in pytorch

Currently, NumPy and CuPy are used for computation on the CPU and GPU, respectively.
This issue will implement the CWT using PyTorch, since it supports both CPU and GPU computation. Moreover, the wavelet parameter (shape factor) can be *optionally learned* when used in a model training pipeline.