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.
  ☆231Updated 7 months ago
• abhiagwl / vistan
  A simple library to run variational inference on Stan models.
  ☆33Updated 2 years ago
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆105Updated 2 years ago
• wesselb / lab
  A generic interface for linear algebra backends
  ☆75Updated last month
• dirmeier / sbijax
  Simulation-based inference in JAX
  ☆42Updated last month
• JakobRobnik / MicroCanonicalHMC
  MCHMC: sampler from an arbitrary differentiable distribution
  ☆75Updated 2 weeks ago
• blackjax-devs / sampling-book
  Tutorials and sampling algorithm comparisons
  ☆79Updated this week
• flatironinstitute / bayes-kit
  Bayesian inference and posterior analysis for Python
  ☆46Updated 2 years ago
• matt-graham / continuously-tempered-hmc
  Continuously tempered Hamiltonian Monte Carlo
  ☆12Updated 8 years ago
• geraschenko / gmrfs
  Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)
  ☆22Updated last year
• aesara-devs / aemcmc
  AeMCMC is a Python library that automates the construction of samplers for Aesara graphs representing statistical models.
  ☆39Updated 2 years ago
• sbi-benchmark / sbibm
  Simulation-based inference benchmark
  ☆107Updated 11 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 last year
• bob-carpenter / walnuts
  Step size adaptation for the No-U-Turn Sampler
  ☆26Updated last month
• aesara-devs / aeppl
  Tools for an Aesara-based PPL.
  ☆67Updated last year
• rlouf / aex
  Sampling with Blackjax on Aesara
  ☆11Updated 2 years ago
• Joshuaalbert / jaxns
  Probabilistic Programming and Nested sampling in JAX
  ☆216Updated last week
• jhuggins / viabel
  Efficient, lightweight variational inference and approximation bounds
  ☆46Updated 2 months ago
• aesara-devs / aehmc
  An HMC/NUTS implementation in Aesara
  ☆31Updated 2 years ago
• justindomke / pangolin
  probabilistic programming focused on fun
  ☆49Updated last week
• bob-carpenter / adaptive-hmc
  Self-tuning HMC algorithms and evaluations
  ☆19Updated last year
• NeilGirdhar / efax
  Exponential families for JAX
  ☆75Updated this week
• danielward27 / flowjax
  ☆205Updated last month
• 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
• flatironinstitute / walnuts
  Within-orbit Adaptive Leapfrog No-U-Turn Sampler
  ☆34Updated last week
• dirmeier / ramsey
  Probabilistic deep learning using JAX
  ☆15Updated 10 months ago
• eigenfoo / littlemcmc
  A lightweight and performant implementation of HMC and NUTS in Python, spun out of the PyMC project.
  ☆59Updated last year
• modichirag / GSM-VI
  Code for Gaussian Score Matching Variational Inference
  ☆35Updated 10 months ago
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆153Updated 3 weeks ago