astro-informatics / s2wav
Differentiable and accelerated wavelet transform on the sphere with JAX
☆17Updated 10 months ago
Alternatives and similar repositories for s2wav:
Users that are interested in s2wav are comparing it to the libraries listed below
- A repository to store state of the art score model architectures☆46Updated 2 months ago
- Learning Diffusion Priors from Observations by Expectation Maximization☆21Updated 2 months ago
- A simple implementation of Hamiltonian Monte Carlo in JAX.☆18Updated last year
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆15Updated 11 months ago
- Stainless neural networks in JAX☆32Updated 4 months ago
- Generative models with JAX.☆17Updated last month
- Implemenation of PQMass from Lemos et al. 2024☆12Updated 3 months ago
- Coverage tests to check the quality of your posterior estimators.☆30Updated 6 months ago
- JAX-powered Cosmological Particle-Mesh N-body Solver☆34Updated 2 weeks ago
- ☆24Updated last year
- JAX-based linear Einstein-Boltzmann solver for cosmology☆14Updated 2 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆89Updated 2 weeks ago
- Differentiable and gpu enabled fast wavelet transforms in JAX.☆43Updated 7 months ago
- Implicit neural representations for strong lensing source reconstruction. Code repository associated with https://arxiv.org/abs/2206.1482…☆11Updated last year
- Simulation-based inference in JAX☆30Updated this week
- Gradient Based Nested Sampling☆19Updated last year
- Probabilistic modeling of tabular data with normalizing flows.☆55Updated 2 weeks ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆65Updated 7 months ago
- JAX bindings for the NVIDIA cuDecomp library☆31Updated last week
- A JAX package for weak gravitational lensing☆30Updated 9 months ago
- nessai: Nested Sampling with Artificial Intelligence☆38Updated 2 weeks ago
- Unitful Quantities in JAX☆30Updated this week
- MCHMC: sampler from an arbitrary differentiable distribution☆72Updated 2 weeks ago
- JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.☆29Updated 2 weeks ago
- Curated list of astronomy/astrophysics code packages using JAX☆37Updated 2 months ago
- A Jax wrapper for cudaKDTree☆10Updated last year
- Numerical quadrature with JAX☆53Updated this week
- a differentiable lensing simulator☆10Updated 3 years ago
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆23Updated 10 months ago