Alternatives and similar repositories for jax_advi

Users that are interested in jax_advi are comparing it to the libraries listed below

• jax-ml / bayeux
  State of the art inference for your bayesian models.
  ☆222Updated 4 months ago
• wesselb / lab
  A generic interface for linear algebra backends
  ☆73Updated 6 months ago
• abhiagwl / vistan
  A simple library to run variational inference on Stan models.
  ☆32Updated 2 years ago
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆105Updated last year
• JakobRobnik / MicroCanonicalHMC
  MCHMC: sampler from an arbitrary differentiable distribution
  ☆74Updated 4 months ago
• flatironinstitute / bayes-kit
  Bayesian inference and posterior analysis for Python
  ☆45Updated last year
• blackjax-devs / sampling-book
  Tutorials and sampling algorithm comparisons
  ☆75Updated this week
• aesara-devs / aeppl
  Tools for an Aesara-based PPL.
  ☆66Updated 10 months ago
• EEA-sensors / sqrt-parallel-smoothers
  A generic library for linear and non-linear Gaussian smoothing problems. The code leverages JAX and implements several linearization algo…
  ☆12Updated 9 months ago
• sbi-benchmark / sbibm
  Simulation-based inference benchmark
  ☆100Updated 7 months ago
• jhuggins / viabel
  Efficient, lightweight variational inference and approximation bounds
  ☆44Updated last year
• sbi-dev / pyknos
  Conditional density estimation with neural networks
  ☆33Updated 7 months ago
• Joshuaalbert / jaxns
  Probabilistic Programming and Nested sampling in JAX
  ☆207Updated last month
• alpiges / LinConGauss
  Integrals of Gaussians under linear domain constraints
  ☆14Updated 5 years ago
• aesara-devs / aehmc
  An HMC/NUTS implementation in Aesara
  ☆31Updated 2 years ago
• bob-carpenter / adaptive-hmc
  Self-tuning HMC algorithms and evaluations
  ☆19Updated 11 months ago
• aesara-devs / aemcmc
  AeMCMC is a Python library that automates the construction of samplers for Aesara graphs representing statistical models.
  ☆40Updated last year
• NeilGirdhar / efax
  Exponential families for JAX
  ☆73Updated this week
• matt-graham / continuously-tempered-hmc
  Continuously tempered Hamiltonian Monte Carlo
  ☆11Updated 8 years ago
• wesselb / varz
  Painless optimisation of constrained variables in AutoGrad, TensorFlow, PyTorch, and JAX
  ☆23Updated 2 years ago
• dirmeier / sbijax
  Simulation-based inference in JAX
  ☆32Updated 5 months ago
• justindomke / pangolin
  probabilistic programming focused on fun
  ☆42Updated last month
• geraschenko / gmrfs
  Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)
  ☆22Updated last year
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆140Updated 3 weeks ago
• rlouf / aex
  Sampling with Blackjax on Aesara
  ☆11Updated 2 years ago
• aterenin / SparseGPAX
  An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.
  ☆22Updated 4 years ago
• pierrejacob / couplingsmontecarlo
  some scripts for the couplings enthusiasts!
  ☆32Updated 5 years ago
• dirmeier / ramsey
  Probabilistic deep learning using JAX
  ☆15Updated 7 months ago
• eigenfoo / littlemcmc
  A lightweight and performant implementation of HMC and NUTS in Python, spun out of the PyMC project.
  ☆57Updated last year
• modichirag / GSM-VI
  Code for Gaussian Score Matching Variational Inference
  ☆35Updated 6 months ago