Differentiable fast wavelet transforms in PyTorch with GPU support.
Differentiable and gpu enabled fast wavelet transforms in JAX.