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