Alternatives and similar repositories for JAX-BO:

Users that are interested in JAX-BO 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
• pierresegonne / VINF
  Repository for DTU Special Course, focusing on Variational Inference using Normalizing Flows (VINF). Supervised by Michael Riis Andersen
  ☆25Updated 4 years ago
• wjmaddox / spectralgp
  Code repo for "Function-Space Distributions over Kernels"
  ☆31Updated 4 years ago
• PrincetonLIPS / AHGP
  [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)
  ☆21Updated 4 years ago
• aterenin / SparseGPAX
  An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.
  ☆22Updated 4 years ago
• karalets / TyXe
  Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)
  ☆42Updated 3 years ago
• willieneis / bayesian-algorithm-execution
  Bayesian algorithm execution (BAX)
  ☆49Updated 3 years ago
• Tom-Ryder / VIforSDEs
  A variational method for fast, approximate inference for stochastic differential equations.
  ☆44Updated 6 years ago
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆102Updated last year
• li012589 / HMC_pytorch
  A pytorch version of hamiltonian monte carlo
  ☆14Updated 5 years ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆44Updated 4 years ago
• johannbrehmer / simulator-mining-example
  Code repository for the generalized Galton board example in the paper "Mining gold from implicit models to improve likelihood-free infere…
  ☆33Updated 5 years ago
• OxfordML / bayesquad
  Library for Bayesian Quadrature
  ☆31Updated 6 years ago
• sremes / nssm-gp
  Non-stationary spectral mixture kernels implemented in GPflow
  ☆28Updated 6 years ago
• probml / JSL
  Jax SSM Library
  ☆49Updated 2 years ago
• gianlucadetommaso / Stein-variational-samplers
  Python and MATLAB code for Stein Variational sampling methods
  ☆25Updated 5 years ago
• j-wilson / GPflowSampling
  Code for efficiently sampling functions from GP(flow) posteriors
  ☆72Updated 4 years ago
• thjashin / solvegp
  Sparse Orthogonal Variational Inference for Gaussian Processes (SOLVE-GP)
  ☆22Updated 3 years ago
• desi-ivanova / idad
  Implicit Deep Adaptive Design (iDAD): Policy-Based Experimental Design without Likelihoods
  ☆19Updated 3 years ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆39Updated 4 years ago
• vr308 / Generalised-Gaussian-Processes
  Fully Bayesian Inference in GPs - Gaussian and Generic Likelihoods
  ☆22Updated last year
• gehring / fax
  ☆80Updated 3 years ago
• google-research / gpax
  ☆30Updated 6 months ago
• secondmind-labs / markovflow
  A Tensorflow based library for Time Series Modelling with Gaussian Processes
  ☆30Updated 9 months ago
• ethanluoyc / dhmc-jax
  Discontinuous Hamiltonian Monte Carlo in JAX
  ☆41Updated 5 years ago
• facebookresearch / rcpm
  Riemannian Convex Potential Maps
  ☆67Updated 2 years ago
• facebookresearch / alebo
  Re-Examining Linear Embeddings for High-dimensional Bayesian Optimization
  ☆41Updated 3 years ago
• juliagusak / neural-ode-norm
  Models and code for the ICLR 2020 workshop paper "Towards Understanding Normalization in Neural ODEs"
  ☆16Updated 4 years ago
• eepperly / Randomly-Pivoted-Cholesky
  Code for the paper "Randomly pivoted Cholesky: Practical approximation of a kernel matrix with few entry evaluations"
  ☆28Updated 6 months ago
• secondmind-labs / bayesian_benchmarks
  A community repository for benchmarking Bayesian methods
  ☆11Updated last year