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
- Gradient Based Nested Sampling☆20Updated last year
- Simulation-based inference in JAX☆31Updated last month
- Generative models with JAX.☆17Updated 2 months ago
- Probabilistic modeling of tabular data with normalizing flows.☆55Updated last month
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆15Updated last year
- Using neural networks to extract sufficient statistics from data by maximising the Fisher information☆31Updated last year
- Things that make me feel productive☆14Updated 2 years ago
- Stainless neural networks in JAX☆33Updated this week
- MCHMC: sampler from an arbitrary differentiable distribution☆71Updated 3 weeks ago
- Sliced Iterative Generator (SIG) & Gaussianizing Iterative Slicing (GIS)☆36Updated 2 years ago
- ☆19Updated 3 years ago
- Code for Gaussian Score Matching Variational Inference☆32Updated last month
- Differentiable and accelerated wavelet transform on the sphere with JAX☆17Updated 11 months ago
- Modern normalizing flows in Python. Simple to use and easily extensible.☆11Updated last week
- Meeting repo for likelihood free inference meeting☆15Updated 2 years ago
- Conditional density estimation with neural networks☆30Updated 2 months ago
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆18Updated last year
- Demonstration of MomentNetworks for high-dimensional probability density estimation (LFI)☆15Updated 2 years ago
- Code for the paper "Towards Reliable Simulation-Based Inference with Balanced Neural Ratio Estimation".☆14Updated 2 years ago
- Combination of transformers and diffusion models for flexible all-in-one simulation-based inference☆60Updated 9 months ago
- Proximal Nested Sampling for high-dimensional Bayesian model selection☆23Updated 11 months ago
- Neural likelihood-free methods in PyTorch.☆39Updated 5 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆102Updated last year
- Normalizing Flows using JAX☆83Updated last year
- ☆49Updated 2 years ago
- Special functions in Jax☆12Updated 3 weeks ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆18Updated 5 months ago
- Next generation Bayesian analysis tools for efficient posterior sampling and evidence estimation.☆29Updated 2 years ago