astro-informatics / s2fft
S2FFT: Differentiable and accelerated spherical transforms
☆150Updated this week
Alternatives and similar repositories for s2fft:
Users that are interested in s2fft are comparing it to the libraries listed below
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆91Updated this week
- A differentiable cosmology library in JAX☆195Updated 4 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆67Updated 2 weeks ago
- Numerical quadrature with JAX☆58Updated last month
- Interpolation and function approximation with JAX☆177Updated 2 weeks ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆233Updated last week
- JAX bindings for the NVIDIA cuDecomp library☆35Updated this week
- Probabilistic Programming and Nested sampling in JAX☆177Updated 3 weeks ago
- Stencil computations in JAX☆70Updated last year
- Unitful Quantities in JAX☆36Updated last week
- JAX + Units☆47Updated last week
- Multiple dispatch over abstract array types in JAX.☆120Updated this week
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Updated last year
- ☆158Updated 2 months ago
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and Optax/JAXopt☆61Updated this week
- A system for scientific simulation-based inference at scale.☆164Updated last year
- Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions☆26Updated last week
- Gravitational-wave data analysis tools in Jax☆68Updated this week
- Evaluate and transform D matrices, 3-j symbols, and (scalar or spin-weighted) spherical harmonics☆47Updated last month
- Orthogonal polynomials with JAX☆21Updated last month
- Automatic Functional Differentiation in JAX☆72Updated last month
- Probabilistic modeling of tabular data with normalizing flows.☆56Updated 2 weeks ago
- Simulation-based inference in JAX☆31Updated 3 weeks ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆29Updated 6 months ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆127Updated 8 months ago
- Minimal Implementation of Bayesian Optimization in JAX☆95Updated 2 weeks ago
- Differentiable Cosmological Forward Model☆79Updated 2 months ago
- JAX-powered Cosmological Particle-Mesh N-body Solver☆40Updated 2 months ago
- Stainless neural networks in JAX☆34Updated last month
- A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control…☆44Updated last month