Joshuaalbert / jaxns
Probabilistic Programming and Nested sampling in JAX
☆150Updated last week
Related projects ⓘ
Alternatives and complementary repositories for jaxns
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆200Updated 2 months ago
- A differentiable cosmology library in JAX☆178Updated 3 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆61Updated 4 months ago
- ☆110Updated 3 weeks ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆81Updated last week
- A system for scientific simulation-based inference at scale.☆161Updated 7 months ago
- Probabilistic modeling of tabular data with normalizing flows.☆55Updated 5 months ago
- Numerical quadrature with JAX☆44Updated 3 weeks ago
- MCHMC: sampler from an arbitrary differentiable distribution☆70Updated this week
- Density Estimation Likelihood-Free Inference with neural density estimators and adaptive acquisition of simulations☆106Updated last year
- Interpolation and function approximation with JAX☆136Updated 3 weeks ago
- The tiniest of Gaussian Process libraries☆297Updated last week
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt☆57Updated this week
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆96Updated last year
- JAX + Units☆43Updated last week
- Simulation-based inference in JAX☆21Updated 2 months ago
- A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work☆21Updated 2 weeks ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆119Updated 3 months ago
- Community-sourced list of papers and resources on neural simulation-based inference.☆96Updated 5 months ago
- Differentiable and accelerated spherical transforms with JAX☆137Updated this week
- Minimal Implementation of Bayesian Optimization in JAX☆84Updated 6 months ago
- Stencil computations in JAX☆66Updated last year
- ⚡️ zeus: Lightning Fast MCMC ⚡️☆227Updated 9 months ago
- Gravitational-wave data analysis tools in Jax☆59Updated 3 weeks ago
- State of the art inference for your bayesian models.☆176Updated last week
- Add a tqdm progress bar to your JAX scans and loops.☆96Updated 3 weeks ago
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆57Updated last week
- Multiple dispatch over abstract array types in JAX.☆106Updated last week
- A dead simple Python package for creating custom JAX pytree objects☆62Updated last week
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆22Updated 7 months ago