jaxwavelets
Extending PyWavelets to JAX. import jaxwavelets as wt.
When to use
Use when the user needs wavelet transforms in JAX. Triggers on: 'wavelet', 'jaxwavelets', 'wavedec', 'waverec', 'dwt', 'idwt', 'cwt', 'swt', 'wavelet packet', 'multiresolution', 'MRA'.
Quick reference
import jaxwavelets as wt
# DWT
cA, cD = wt.dwt(x, 'db4')
x_rec = wt.idwt(cA, cD, 'db4')
# Multilevel nD
coeffs = wt.wavedecn(x, 'db4', level=3)
x_rec = wt.waverecn(coeffs, 'db4')
# 2D convenie
[Description truncada. Veja o README completo no GitHub.]