astro-informatics / s2wavLinks
Differentiable and accelerated wavelet transform on the sphere with JAX
☆23Updated 4 months ago
Alternatives and similar repositories for s2wav
Users that are interested in s2wav are comparing it to the libraries listed below
Sorting:
- Stainless neural networks in JAX☆34Updated last month
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆101Updated this week
- Simulation-based inference in JAX☆35Updated 5 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆74Updated last week
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆19Updated last year
- JAX bindings for the NVIDIA cuDecomp library☆44Updated 2 weeks ago
- Fast, lightweight and parallelised simulation-based inference in JAX.☆19Updated last week
- A repository to store state of the art score model architectures