Extending PyWavelets to JAX. `import jaxwavelets as wt`.
Scanned 5/27/2026
Install via CLI
openskills install handley-lab/jaxwavelets# 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
```python
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 convenience
coeffs = wt.wavedec2(x, 'db4', level=3)
x_rec = wt.waverec2(coeffs, 'db4')
# SWT (shift-invariant)
coeffs = wt.swt(x, 'db4', level=3)
x_rec = wt.iswt(coeffs, 'db4')
# CWT (two-phase for JIT)
bank = wt.prepare_cwt(scales, 'morl')
coefs, freqs = wt.apply_cwt(data, bank) # JIT-safe
# Composability
import jax
from functools import partial
jax.vmap(partial(wt.wavedecn, wavelet='db4', level=3))(batch)
jax.grad(lambda x: jnp.sum(wt.waverecn(wt.wavedecn(x, 'db4'), 'db4')))(x)
jax.jit(wt.wavedecn, static_argnames=['wavelet', 'mode', 'level'])(x, wavelet='db4')
```
## Available wavelets
- **Discrete:** haar, db1-20, sym2-20, coif1-5
- **Continuous:** morl, mexh, gaus1-8, cgau1-8, cmor, shan, fbsp
## Key design points
- `import jaxwavelets as wt`
- Single-example functions — user applies `vmap`/`pmap`/`grad`/`jit` from outside
- No complex arithmetic internally (GPU-safe)
- No numpy dependency at runtime
- Coefficients are JAX pytrees (`WaveletCoeffs`, `CWTKernelBank`)
- Modes: `'symmetric'`, `'reflect'`, `'periodization'`
- CWT needs `prepare_cwt()` outside JIT, `apply_cwt()` inside JIT
No comments yet. Be the first to comment!
This skill helps you track, analyze, and report on keyword ranking positions over time. It monitors both traditional SERP rankings and AI/GEO visibility to provide comprehensive search performance insights.
Find and analyze YouTube competitor channels using YouTube Data API v3. Discover competitors through keyword search, category matching, content similarity, and related channel discovery. Compare metrics, content strategies, and market positioning. Use when users want to (1) Find competitors for their YouTube channel, (2) Analyze competitor performance metrics, (3) Compare their channel against competitors, (4) Identify content gaps and opportunities, (5) Benchmark against similar creators, (6...
Get current weather and forecasts (no API key required).
Analyze and optimize tweets for maximum reach using Twitter's open-source algorithm insights. Rewrite and edit user tweets to improve engagement and visibility based on how the recommendation system ranks content.
Master dbt (data build tool) for analytics engineering with model organization, testing, documentation, and incremental strategies. Use when building data transformations, creating data models, or implementing analytics engineering best practices.