martin-marek / mini-hmc-jax
A simple implementation of Hamiltonian Monte Carlo in JAX.
☆18Updated last year
Alternatives and similar repositories for mini-hmc-jax:
Users that are interested in mini-hmc-jax are comparing it to the libraries listed below
- ☆24Updated last year
- Generative models with JAX.☆17Updated last month
- Probabilistic modeling of tabular data with normalizing flows.☆55Updated 2 weeks ago
- Gradient Based Nested Sampling☆19Updated last year
- Simulation-based inference in JAX☆30Updated this week
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆15Updated 11 months ago
- Sliced Iterative Generator (SIG) & Gaussianizing Iterative Slicing (GIS)☆36Updated last year
- MCHMC: sampler from an arbitrary differentiable distribution☆72Updated 2 weeks ago
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆31Updated last year
- Stainless neural networks in JAX☆32Updated 4 months ago
- Differentiable and accelerated wavelet transform on the sphere with JAX☆17Updated 10 months ago
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆23Updated 10 months ago
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆17Updated last year
- Things that make me feel productive☆15Updated 2 years ago
- Demonstration of MomentNetworks for high-dimensional probability density estimation (LFI)☆14Updated 2 years ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆16Updated 3 months ago
- Code for Gaussian Score Matching Variational Inference☆32Updated 4 months ago
- Using Graph Neural Networks to regress baryonic properties directly from full dark matter merger trees.☆24Updated last year
- Normalizing Flows using JAX☆82Updated last year
- Coverage tests to check the quality of your posterior estimators.☆30Updated 6 months ago
- Diagnostics for Conditional Density Estimators and Bayesian Inference Algorithms☆14Updated 2 years ago
- Meeting repo for likelihood free inference meeting☆15Updated 2 years ago
- ☆16Updated 4 years ago
- Hierarchical neural implicit inference over event ensembles. Code repository associated with https://arxiv.org/abs/2306.12584.☆13Updated last year
- ☆49Updated last year
- Stencil computations in JAX☆70Updated last year
- Conditional density estimation with neural networks☆30Updated last month
- nessai: Nested Sampling with Artificial Intelligence☆38Updated 2 weeks ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆100Updated last year