Differentiable and accelerated wavelet transform on the sphere with JAX
☆25May 27, 2025Updated 10 months ago
Alternatives and similar repositories for s2wav
Users that are interested in s2wav are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Fast and exact spin spherical harmonic transforms☆12Apr 1, 2025Updated last year
- PyTorch-based radio-interferometric imaging reconstruction package with scalable Bayesian uncertainty quantification relying on data-driv…☆12Feb 17, 2025Updated last year
- Differentiable and GPU accelerated scattering covariance statistics on the sphere☆15May 27, 2025Updated 10 months ago
- Fast wavelet transforms on the sphere.☆22Mar 30, 2025Updated last year
- WavSpA: Wavelet Space Attention for Enhancing Transformer's Long Sequence Learning☆12Feb 24, 2024Updated 2 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆79Mar 18, 2026Updated last month
- A TensorFlow reimplementation of GalSim☆10Mar 8, 2022Updated 4 years ago
- simple simulations for testing weak lensing shear measurement☆13Updated this week
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆13Dec 12, 2022Updated 3 years ago
- Stainless neural networks in JAX☆34Feb 3, 2026Updated 2 months ago
- JAX Tutorial notebooks : basics, crash & tips, usage of optax/JaxOptim/Numpyro☆16Feb 25, 2025Updated last year
- JAX port of GalSim, for parallelized, GPU accelerated, and differentiable galaxy image simulations.☆33Updated this week
- ☆13Jan 3, 2025Updated last year
- generate realistic atmospheric input parameters for point-spread function simulations in GalSim☆16Oct 29, 2024Updated last year
- Deploy open-source AI quickly and easily - Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- JAX-powered Cosmological Particle-Mesh N-body Solver☆63Mar 31, 2026Updated 2 weeks ago
- The galaxy survey codex☆17Jul 12, 2024Updated last year
- Modern normalizing flows in Python. Simple to use and easily extensible.☆12Mar 9, 2026Updated last month
- Shape Measurement Pipeline☆17Mar 31, 2026Updated 2 weeks ago
- Curated list of astronomy/astrophysics code packages using JAX☆54Mar 25, 2026Updated 3 weeks ago
- Solving elliptic differential equations using neural networks in plasma fluid simulations. Mirror repository of https://gitlab.com/cerfac…☆13May 6, 2022Updated 3 years ago
- The open-source code of "CS-SHAP: Extending SHAP to Cyclic-Spectral Domain for Better Interpretability of Intelligent Fault Diagnosis".☆20Jul 12, 2025Updated 9 months ago
- Wavelet Phase Harmonics in PyTorch for Images☆20Jan 28, 2025Updated last year
- Differentiable Cosmological Forward Model☆91Apr 8, 2026Updated last week
- Deploy open-source AI quickly and easily - Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- A differentiable cosmology library in JAX☆233Jun 27, 2025Updated 9 months ago
- Code for Gaussian Score Matching Variational Inference☆34Feb 23, 2025Updated last year
- Differentiable Spherical Harmonic Transforms in PyTorch☆17Sep 19, 2022Updated 3 years ago
- Differentiable and gpu enabled fast wavelet transforms in JAX.☆48Jul 5, 2024Updated last year
- Next-generation radio interferometric imaging.☆20Apr 10, 2025Updated last year
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆75Feb 12, 2026Updated 2 months ago
- A PyTorch implementation of the CorefQA Model.☆10Jun 27, 2020Updated 5 years ago
- ☆16Nov 3, 2022Updated 3 years ago
- (Reduced) Wavelet Scattering Transform on Images☆13Jan 28, 2025Updated last year
- Serverless GPU API endpoints on Runpod - Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Implementations of orthogonal and semi-orthogonal convolutions in the Fourier domain with applications to adversarial robustness☆48Apr 9, 2021Updated 5 years ago
- A framework for creating databases for surrogate models of complex physics codes.☆12Updated this week
- ☆22May 6, 2020Updated 5 years ago
- S2FFT: Differentiable and accelerated spherical transforms☆223Apr 11, 2026Updated last week
- Implemenation of PQMass from Lemos et al. 2024☆20Apr 23, 2025Updated 11 months ago
- Gradient Based Nested Sampling☆21Feb 2, 2024Updated 2 years ago
- Next generation Bayesian analysis tools for efficient posterior sampling and evidence estimation.☆27Oct 13, 2022Updated 3 years ago