Alternatives and similar repositories for jax_advi:

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

• sbi-dev / pyknos
  Conditional density estimation with neural networks
  ☆30Updated 2 months ago
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆101Updated last year
• aterenin / SparseGPAX
  An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.
  ☆22Updated 3 years ago
• JakobRobnik / MicroCanonicalHMC
  MCHMC: sampler from an arbitrary differentiable distribution
  ☆71Updated 2 weeks ago
• geraschenko / gmrfs
  Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)
  ☆20Updated 11 months ago
• wesselb / lab
  A generic interface for linear algebra backends
  ☆71Updated 2 weeks 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.
  ☆204Updated 3 months ago
• NeilGirdhar / efax
  Exponential families for JAX
  ☆63Updated last week
• sbi-benchmark / sbibm
  Simulation-based inference benchmark
  ☆96Updated last month
• flatironinstitute / bayes-kit
  Bayesian inference and posterior analysis for Python
  ☆44Updated last year
• dirmeier / sbijax
  Simulation-based inference in JAX
  ☆31Updated last month
• modichirag / GSM-VI
  Code for Gaussian Score Matching Variational Inference
  ☆32Updated 3 weeks ago
• AdrienCorenflos / particle_mala
  Gradient-informed particle MCMC methods
  ☆11Updated last year
• google-deepmind / annealed_flow_transport
  ☆49Updated 2 years ago
• justindomke / pangolin
  probabilistic programming focused on fun
  ☆40Updated 4 months ago
• 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 7 months ago
• bob-carpenter / adaptive-hmc
  Self-tuning HMC algorithms and evaluations
  ☆18Updated 5 months ago
• alpiges / LinConGauss
  Integrals of Gaussians under linear domain constraints
  ☆13Updated 5 years ago
• eigenfoo / littlemcmc
  A lightweight and performant implementation of HMC and NUTS in Python, spun out of the PyMC project.
  ☆53Updated 7 months ago
• vr308 / Generalised-Gaussian-Processes
  Fully Bayesian Inference in GPs - Gaussian and Generic Likelihoods
  ☆22Updated last year
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆41Updated last week
• markvdw / RobustGP
  Robust initialisation of inducing points in sparse variational GP regression models.
  ☆33Updated 2 years ago
• rlouf / aex
  Sampling with Blackjax on Aesara
  ☆11Updated 2 years ago
• aterenin / SparseGaussianProcesses.jl
  A Julia implementation of sparse Gaussian processes via path-wise doubly stochastic variational inference.
  ☆33Updated 4 years ago
• probabilistic-numerics / probnum-spring-school
  Tutorial materials of the Probabilistic Numerics Spring School.
  ☆34Updated last year
• aesara-devs / aehmc
  An HMC/NUTS implementation in Aesara
  ☆32Updated last year
• jhuggins / viabel
  Efficient, lightweight variational inference and approximation bounds
  ☆43Updated last year
• blackjax-devs / sampling-book
  Tutorials and sampling algorithm comparisons
  ☆71Updated this week