Joshuaalbert / jaxns
Probabilistic Programming and Nested sampling in JAX
☆147Updated this week
Related projects ⓘ
Alternatives and complementary repositories for jaxns
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆199Updated last month
- A differentiable cosmology library in JAX☆176Updated 3 months ago
- MCHMC: sampler from an arbitrary differentiable distribution☆70Updated this week
- A system for scientific simulation-based inference at scale.☆161Updated 7 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆61Updated 4 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆80Updated this week
- ☆101Updated 2 weeks ago
- Numerical quadrature with JAX☆42Updated 2 weeks ago
- Probabilistic modeling of tabular data with normalizing flows.☆55Updated 4 months ago
- Machine learning assisted marginal likelihood (Bayesian evidence) estimation for Bayesian model selection☆56Updated this week
- Density Estimation Likelihood-Free Inference with neural density estimators and adaptive acquisition of simulations☆104Updated last year
- Interpolation and function approximation with JAX☆131Updated last week
- The tiniest of Gaussian Process libraries☆295Updated this week
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt☆57Updated 3 weeks ago
- Gravitational-wave data analysis tools in Jax☆58Updated last week
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆95Updated last year
- A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work☆21Updated this week
- Simulation-based inference in JAX☆21Updated last month
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆22Updated 6 months ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆119Updated 2 months ago
- Stencil computations in JAX☆66Updated last year
- Community-sourced list of papers and resources on neural simulation-based inference.☆92Updated 4 months ago
- Differentiable and accelerated spherical transforms with JAX☆133Updated 2 weeks ago
- Simulation-based inference benchmark☆89Updated 7 months ago
- State of the art inference for your bayesian models.☆163Updated last month
- JAX + Units☆43Updated this week
- nessai: Nested Sampling with Artificial Intelligence☆36Updated this week
- A dead simple Python package for creating custom JAX pytree objects☆61Updated 4 months ago
- Differentiable Gravitational Waveforms with JAX☆54Updated 2 months ago
- Minimal Implementation of Bayesian Optimization in JAX☆84Updated 6 months ago