Differentiable and accelerated wavelet transform on the sphere with JAX
☆25May 27, 2025Updated 9 months ago
Alternatives and similar repositories for s2wav
Users that are interested in s2wav are comparing it to the libraries listed below
Sorting:
- PyTorch-based radio-interferometric imaging reconstruction package with scalable Bayesian uncertainty quantification relying on data-driv…☆12Feb 17, 2025Updated last year
- Fast and exact spin spherical harmonic transforms☆12Apr 1, 2025Updated 11 months ago
- WavSpA: Wavelet Space Attention for Enhancing Transformer's Long Sequence Learning☆12Feb 24, 2024Updated 2 years ago
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆12Dec 12, 2022Updated 3 years ago
- generate realistic atmospheric input parameters for point-spread function simulations in GalSim☆16Oct 29, 2024Updated last year
- A TensorFlow reimplementation of GalSim☆10Mar 8, 2022Updated 3 years ago
- simple simulations for testing weak lensing shear measurement☆12Oct 31, 2025Updated 4 months ago
- JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.☆33Feb 22, 2026Updated last week
- Stainless neural networks in JAX☆34Feb 3, 2026Updated last month
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆77Feb 18, 2026Updated 2 weeks ago
- The galaxy survey codex☆17Jul 12, 2024Updated last year
- Shape Measurement Pipeline☆17Feb 13, 2026Updated 3 weeks ago
- JAX Tutorial notebooks : basics, crash & tips, usage of optax/JaxOptim/Numpyro☆16Feb 25, 2025Updated last year
- Curated list of astronomy/astrophysics code packages using JAX☆52Feb 18, 2026Updated 2 weeks ago
- JAX-powered Cosmological Particle-Mesh N-body Solver☆59Feb 24, 2026Updated last week
- ☆22May 6, 2020Updated 5 years ago
- c++ backed python optical raytracer☆26Feb 12, 2025Updated last year
- Coverage tests to check the quality of your posterior estimators.☆33Aug 12, 2024Updated last year
- Code for Gaussian Score Matching Variational Inference☆35Feb 23, 2025Updated last year
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆75Feb 12, 2026Updated 3 weeks ago
- Differentiable cosmological emulators: the JAX version of CosmoPower☆44Nov 27, 2025Updated 3 months ago
- A differentiable cosmology library in JAX☆225Jun 27, 2025Updated 8 months ago
- Differentiable models of halo and galaxy formation history☆10Updated this week
- Pseudospectral PDE solver☆12Jun 24, 2025Updated 8 months ago
- Diffusion Probabilistic Model in Jax☆12Apr 20, 2024Updated last year
- Offline Particle Tracking Project for ROMS output. Based on LTRANS☆10Dec 10, 2025Updated 2 months ago
- Weak lensing measurement pipeline for wide area surveys☆14Apr 14, 2022Updated 3 years ago
- Differentiable Cosmological Forward Model☆90Feb 26, 2026Updated last week
- Implementations of orthogonal and semi-orthogonal convolutions in the Fourier domain with applications to adversarial robustness☆48Apr 9, 2021Updated 4 years ago
- Next generation Bayesian analysis tools for efficient posterior sampling and evidence estimation.☆27Oct 13, 2022Updated 3 years ago
- This project compares the performance of Swin-Transformer v2 implemented in JAX and PyTorch.☆12Jun 8, 2022Updated 3 years ago
- Preconditioned forward/backward clean algorithm☆11Feb 27, 2026Updated last week
- Python GUI for differential forms☆13Oct 14, 2023Updated 2 years ago
- Quantum simulator of qudits☆18Aug 19, 2025Updated 6 months ago
- Gramm calculation core☆11Feb 27, 2026Updated last week
- The ERSN-OpenMC-Py is a graphical user interface specifically designed to streamline the use of the OpenMC code. It provides a user-frie…☆11Feb 24, 2024Updated 2 years ago
- Modern normalizing flows in Python. Simple to use and easily extensible.☆12Feb 11, 2026Updated 3 weeks ago
- Code to create gravitational-wave memory waveforms☆10Jul 14, 2024Updated last year
- Kernel Localized Linear Regression (KLLR)☆11Jan 31, 2024Updated 2 years ago