Alternatives and similar repositories for jax_advi:

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

• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆100Updated last year
• geraschenko / gmrfs
  Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)
  ☆19Updated 9 months ago
• JakobRobnik / MicroCanonicalHMC
  MCHMC: sampler from an arbitrary differentiable distribution
  ☆72Updated last week
• abhiagwl / vistan
  A simple library to run variational inference on Stan models.
  ☆30Updated last year
• modichirag / GSM-VI
  Code for Gaussian Score Matching Variational Inference
  ☆31Updated 4 months ago
• wesselb / lab
  A generic interface for linear algebra backends
  ☆71Updated 7 months ago
• aesara-devs / aemcmc
  AeMCMC is a Python library that automates the construction of samplers for Aesara graphs representing statistical models.
  ☆39Updated last year
• jax-ml / bayeux
  State of the art inference for your bayesian models.
  ☆198Updated 2 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 2 months ago
• justindomke / pangolin
  probabilistic programming focused on fun
  ☆39Updated 3 months ago
• sbi-dev / pyknos
  Conditional density estimation with neural networks
  ☆30Updated 3 weeks ago
• dirmeier / sbijax
  Simulation-based inference in JAX
  ☆29Updated last week
• NeilGirdhar / efax
  Exponential families for JAX
  ☆63Updated this week
• blackjax-devs / sampling-book
  Tutorials and sampling algorithm comparisons
  ☆70Updated this week
• flatironinstitute / bayes-kit
  Bayesian inference and posterior analysis for Python
  ☆44Updated last year
• danielward27 / flowjax
  ☆141Updated this week
• pnkraemer / tornadox
  Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…
  ☆20Updated 6 months ago
• aterenin / SparseGPAX
  An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.
  ☆22Updated 3 years ago
• AdrienCorenflos / particle_mala
  Gradient-informed particle MCMC methods
  ☆11Updated last year
• aesara-devs / aeppl
  Tools for an Aesara-based PPL.
  ☆64Updated 3 months ago
• bob-carpenter / adaptive-hmc
  Self-tuning HMC algorithms and evaluations
  ☆18Updated 4 months ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆41Updated 3 months ago
• probabilistic-numerics / probnum-spring-school
  Tutorial materials of the Probabilistic Numerics Spring School.
  ☆34Updated last year
• markvdw / RobustGP
  Robust initialisation of inducing points in sparse variational GP regression models.
  ☆33Updated 2 years ago
• sbi-benchmark / sbibm
  Simulation-based inference benchmark
  ☆95Updated 3 weeks ago
• aterenin / SparseGaussianProcesses.jl
  A Julia implementation of sparse Gaussian processes via path-wise doubly stochastic variational inference.
  ☆33Updated 4 years ago
• alpiges / LinConGauss
  Integrals of Gaussians under linear domain constraints
  ☆13Updated 4 years ago
• aesara-devs / aehmc
  An HMC/NUTS implementation in Aesara
  ☆32Updated last year
• mackelab / simformer
  Combination of transformers and diffusion models for flexible all-in-one simulation-based inference
  ☆56Updated 7 months ago
• eigenfoo / littlemcmc
  A lightweight and performant implementation of HMC and NUTS in Python, spun out of the PyMC project.
  ☆53Updated 6 months ago