martin-marek / mini-hmc-jax
A simple implementation of Hamiltonian Monte Carlo in JAX.
☆17Updated 11 months ago
Alternatives and similar repositories for mini-hmc-jax:
Users that are interested in mini-hmc-jax are comparing it to the libraries listed below
- Generative models with JAX.☆17Updated 2 weeks ago
- ☆21Updated last year
- Gradient Based Nested Sampling☆19Updated 11 months ago
- Simulation-based inference in JAX☆29Updated this week
- Probabilistic modeling of tabular data with normalizing flows.☆55Updated 7 months ago
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆31Updated last year
- MCHMC: sampler from an arbitrary differentiable distribution☆72Updated this week
- Stainless neural networks in JAX☆32Updated 3 months ago
- Code for Gaussian Score Matching Variational Inference☆28Updated 3 months ago
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆23Updated 9 months ago
- Sliced Iterative Generator (SIG) & Gaussianizing Iterative Slicing (GIS)☆36Updated last year
- Normalizing Flows using JAX☆82Updated last year
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆12Updated 10 months ago
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆17Updated last year
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆99Updated last year
- Demonstration of MomentNetworks for high-dimensional probability density estimation (LFI)☆14Updated 2 years ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆10Updated 2 months ago
- Neural likelihood-free methods in PyTorch.☆39Updated 4 years ago
- Conditional density estimation with neural networks☆29Updated 4 months ago
- ☆129Updated 3 weeks ago
- ☆49Updated last year
- Meeting repo for likelihood free inference meeting☆15Updated 2 years ago
- Combination of transformers and diffusion models for flexible all-in-one simulation-based inference☆52Updated 7 months ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆121Updated 4 months ago
- A dead simple Python package for creating custom JAX pytree objects☆62Updated 2 months ago
- Differentiable and accelerated wavelet transform on the sphere with JAX☆17Updated 9 months ago
- Things that make me feel productive☆15Updated 2 years ago
- nessai: Nested Sampling with Artificial Intelligence☆37Updated this week
- Coverage tests to check the quality of your posterior estimators.☆29Updated 5 months ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆43Updated this week