v0lta / Jax-Wavelet-ToolboxLinks
Differentiable and gpu enabled fast wavelet transforms in JAX.
☆41Updated 11 months ago
Alternatives and similar repositories for Jax-Wavelet-Toolbox
Users that are interested in Jax-Wavelet-Toolbox are comparing it to the libraries listed below
Sorting:
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆95Updated this week
- Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions☆30Updated last week
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆72Updated last month
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 4 months ago
- Stainless neural networks in JAX