Alternatives and similar repositories for VAI-Lab:

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

• wesselb / lab
  A generic interface for linear algebra backends
  ☆71Updated 7 months ago
• PrincetonLIPS / AHGP
  [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)
  ☆21Updated 4 years ago
• aidanscannell / GPJax
  Minimal Gaussian process library in JAX with a simple (custom) approach to state management.
  ☆11Updated last year
• lbl-camera / fvGP
  A software package for flexible HPC GPs
  ☆15Updated 3 weeks ago
• cambridge-mlg / sgd-gp
  Public code for running Stochastic Gradient Descent on GPs.
  ☆36Updated 3 months ago
• justindomke / pangolin
  probabilistic programming focused on fun
  ☆39Updated 3 months ago
• aterenin / SparseGPAX
  An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.
  ☆22Updated 3 years 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 3 months ago
• operalib / operalib
  Learning with operator-valued kernels
  ☆22Updated 2 years ago
• google-research / gpax
  ☆30Updated 5 months ago
• yucenli / bnn-bo
  Bayesian Neural Network Surrogates for Bayesian Optimization
  ☆48Updated 9 months ago
• amzn / orthogonal-additive-gaussian-processes
  Light-weighted code for Orthogonal Additive Gaussian Processes
  ☆41Updated 6 months ago
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆114Updated last month
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated last year
• IdahoLabResearch / BIhNNs
  The code enables to perform Bayesian inference in an efficient manner through the use of Hamiltonian Neural Networks (HNNs), Deep Neural …
  ☆14Updated 2 years ago
• JonathanWenger / itergp
  IterGP: Computation-Aware Gaussian Process Inference (NeurIPS 2022)
  ☆40Updated last year
• NeilGirdhar / efax
  Exponential families for JAX
  ☆63Updated last week
• marvinpfoertner / linpde-gp
  Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"
  ☆25Updated 9 months ago
• MJHutchinson / ExtrinsicGaugeIndependentVectorGPs
  A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes
  ☆25Updated 3 years ago
• PredictiveIntelligenceLab / JAX-BO
  ☆30Updated 2 years ago
• runame / laplace-redux
  Laplace Redux -- Effortless Bayesian Deep Learning
  ☆41Updated last year
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆49Updated 3 years ago
• alonfnt / bayex
  Minimal Implementation of Bayesian Optimization in JAX
  ☆88Updated last month
• j-wilson / GPflowSampling
  Code for efficiently sampling functions from GP(flow) posteriors
  ☆69Updated 4 years ago
• spbu-math-cs / Riemannian-Gaussian-Processes
  Supplementary code for the NeurIPS 2020 paper "Matern Gaussian processes on Riemannian manifolds".
  ☆28Updated 2 weeks ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆45Updated 2 weeks ago
• brandstetter-johannes / LPSDA
  Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"
  ☆49Updated last year
• google-deepmind / annealed_flow_transport
  ☆49Updated last year