f0uriest / quadax
Numerical quadrature with JAX
☆39Updated last month
Related projects: ⓘ
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆77Updated last week
- Interpolation and function approximation with JAX☆117Updated 2 weeks ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆61Updated 2 months ago
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt☆45Updated this week
- MCHMC: sampler from an arbitrary differentiable distribution☆69Updated this week
- Probabilistic Programming and Nested sampling in JAX☆141Updated 3 weeks ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆188Updated last week
- JAX bindings for the NVIDIA cuDecomp library☆24Updated 2 weeks ago
- Differentiable and accelerated spherical transforms with JAX☆123Updated this week
- JAX + Units☆41Updated last week
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆10Updated 6 months ago
- ☆31Updated 3 years ago
- A differentiable cosmology library in JAX☆172Updated last month
- Simulation-based inference in JAX☆18Updated last month
- JAX-powered Cosmological Particle-Mesh N-body Solver☆23Updated last month
- Stencil computations in JAX☆66Updated 11 months ago
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆20Updated 5 months ago
- ☆85Updated this week
- Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions☆16Updated this week
- Differentiable Gravitational Waveforms with JAX☆53Updated 3 weeks ago
- Differentiable Cosmological Forward Model☆69Updated 2 weeks ago
- A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work☆18Updated last week
- A dead simple Python package for creating custom JAX pytree objects☆62Updated 2 months ago
- Next generation Bayesian analysis tools for efficient posterior sampling and evidence estimation.☆29Updated last year
- Evaluate and transform D matrices, 3-j symbols, and (scalar or spin-weighted) spherical harmonics☆45Updated 9 months ago
- Density Estimation Likelihood-Free Inference with neural density estimators and adaptive acquisition of simulations☆102Updated last year
- Implementation of Arxiv:2212.08549 in Julia☆11Updated this week
- Tool to distribute a list of computational tasks over a pool of compute resources. The pool can grow or shrink.☆39Updated 5 months ago
- A spherical convolutional neural network for cosmology (TFv2).☆19Updated 5 months ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆47Updated 7 months ago