Alternatives and similar repositories for VAI-Lab

Users that are interested in VAI-Lab are comparing it to the libraries listed below

• aidanscannell / GPJax
  Minimal Gaussian process library in JAX with a simple (custom) approach to state management.
  ☆11Updated last year
• cambridge-mlg / sgd-gp
  Public code for running Stochastic Gradient Descent on GPs.
  ☆38Updated 3 weeks ago
• lbl-camera / fvGP
  A software package for flexible HPC GPs
  ☆16Updated this week
• aterenin / SparseGPAX
  An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.
  ☆22Updated 4 years ago
• wesselb / lab
  A generic interface for linear algebra backends
  ☆73Updated 2 months ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆46Updated last month
• marvinpfoertner / linpde-gp
  Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"
  ☆26Updated last year
• vr308 / Generalised-Gaussian-Processes
  Fully Bayesian Inference in GPs - Gaussian and Generic Likelihoods
  ☆22Updated last year
• ylefay / bayesianSDEsolver
  Efficient SDE samplers including Gaussian-based probabilistic solvers. Written in JAX.
  ☆10Updated 3 months ago
• PrincetonLIPS / AHGP
  [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)
  ☆21Updated 4 years ago
• MJHutchinson / ExtrinsicGaugeIndependentVectorGPs
  A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes
  ☆25Updated 3 years ago
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆118Updated 4 months ago
• justindomke / pangolin
  probabilistic programming focused on fun
  ☆40Updated 6 months ago
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆49Updated 3 years ago
• acerbilab / amortized-conditioning-engine
  Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)
  ☆16Updated 3 weeks ago
• j-wilson / GPflowSampling
  Code for efficiently sampling functions from GP(flow) posteriors
  ☆72Updated 4 years ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• operalib / operalib
  Learning with operator-valued kernels
  ☆22Updated 2 years ago
• desResLab / LINFA
  A Python library for variational inference with normalizing flow and annealing
  ☆16Updated 2 months ago
• ae-foster / dad
  Deep Adaptive Design: Amortizing Sequential Bayesian Experimental Design
  ☆31Updated 3 years ago
• google-research / gpax
  ☆30Updated 8 months ago
• ramsey-devs / ramsey
  Probabilistic deep learning using JAX
  ☆13Updated 3 months ago
• pierresegonne / VINF
  Repository for DTU Special Course, focusing on Variational Inference using Normalizing Flows (VINF). Supervised by Michael Riis Andersen
  ☆25Updated 4 years ago
• cagatayyildiz / npde
  Nonparametric Differential Equation Modeling
  ☆53Updated last year
• flowersteam / sbmltoodejax
  ☆11Updated 4 months ago
• ericmjl / score-models
  In which I learn about score functions and how they can be used to generate data.
  ☆16Updated last year
• yucenli / bnn-bo
  Bayesian Neural Network Surrogates for Bayesian Optimization
  ☆51Updated last year
• vdutor / SphericalHarmonics
  Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax
  ☆29Updated last year
• 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…
  ☆21Updated 9 months ago
• f-dangel / phd-thesis
  Source code for my PhD thesis: Backpropagation Beyond the Gradient
  ☆20Updated 2 years ago