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 6 months ago
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆113Updated last week
• justindomke / pangolin
  probabilistic programming focused on fun
  ☆39Updated 2 months ago
• cambridge-mlg / sgd-gp
  Public code for running Stochastic Gradient Descent on GPs.
  ☆36Updated 2 months ago
• lbl-camera / fvGP
  A software package for flexible HPC GPs
  ☆14Updated last month
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆49Updated 3 years ago
• aidanscannell / GPJax
  Minimal Gaussian process library in JAX with a simple (custom) approach to state management.
  ☆11Updated last year
• vr308 / Generalised-Gaussian-Processes
  Fully Bayesian Inference in GPs - Gaussian and Generic Likelihoods
  ☆22Updated last year
• marvinpfoertner / linpde-gp
  Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"
  ☆24Updated 8 months ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆43Updated last week
• Bostanabad-Research-Group / GP-Plus
  Python Library for Generalized Gaussian Process Modeling
  ☆21Updated 6 months ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 3 years ago
• probabilistic-numerics / probnum-spring-school
  Tutorial materials of the Probabilistic Numerics Spring School.
  ☆34Updated last year
• aterenin / SparseGPAX
  An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.
  ☆22Updated 3 years ago
• vdutor / SphericalHarmonics
  Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax
  ☆27Updated 8 months ago
• google-research / gpax
  ☆28Updated 4 months ago
• yzshi5 / OpFlow
  codes for "Universal Functional Regression with Neural Operator Flows"
  ☆13Updated 3 months ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆38Updated 2 months ago
• lockwo / distreqx
  Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
  ☆24Updated last month
• j-wilson / GPflowSampling
  Code for efficiently sampling functions from GP(flow) posteriors
  ☆68Updated 4 years ago
• JonathanWenger / itergp
  IterGP: Computation-Aware Gaussian Process Inference (NeurIPS 2022)
  ☆40Updated last year
• imbirik / LieStationaryKernels
  Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"
  ☆42Updated last year
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆48Updated last year
• PrincetonLIPS / AHGP
  [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)
  ☆21Updated 4 years ago
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆99Updated last year
• alan-turing-institute / stat-fem
  Python tools for solving data-constrained finite element problems
  ☆12Updated 3 years ago
• MJHutchinson / ExtrinsicGaugeIndependentVectorGPs
  A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes
  ☆25Updated 3 years ago
• google-research / hyperbo
  Pre-trained Gaussian processes for Bayesian optimization
  ☆88Updated last month
• blackjax-devs / sampling-book
  Tutorials and sampling algorithm comparisons
  ☆70Updated this week
• alonfnt / bayex
  Minimal Implementation of Bayesian Optimization in JAX
  ☆88Updated last week