PredictiveIntelligenceLab / JAX-BOLinks
☆30Updated 3 years ago
Alternatives and similar repositories for JAX-BO
Users that are interested in JAX-BO are comparing it to the libraries listed below
Sorting:
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆12Updated last year
- Repository for DTU Special Course, focusing on Variational Inference using Normalizing Flows (VINF). Supervised by Michael Riis Andersen☆25Updated 5 years ago
- Nonparametric Differential Equation Modeling☆54Updated last year
- Deep GPs built on top of TensorFlow/Keras and GPflow☆127Updated 11 months ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Code for efficiently sampling functions from GP(flow) posteriors☆73Updated 4 years ago
- ☆110Updated 4 years ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Updated 4 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Updated last year
- Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)☆42Updated 4 years ago
- Jax SSM Library☆49Updated 2 years ago
- A Tensorflow based library for Time Series Modelling with Gaussian Processes☆32Updated last year
- Bayesian algorithm execution (BAX)☆50Updated 4 years ago
- simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…☆15Updated 4 years ago
- A variational method for fast, approximate inference for stochastic differential equations.☆44Updated 7 years ago
- Parametric Gaussian Process Regression for Big Data☆45Updated 5 years ago
- Normalizing Flows using JAX☆84Updated last year
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆100Updated 2 years ago
- ☆29Updated 2 years ago
- Code repo for "Function-Space Distributions over Kernels"☆32Updated 4 years ago
- A curated list of resources for learning Gaussian Processes☆41Updated 4 years ago
- Neat Bayesian machine learning examples☆58Updated last week
- Layered distributions using FLAX/JAX☆10Updated 4 years ago
- Documentation:☆123Updated 2 years ago
- Bayesian inference with Python and Jax.☆34Updated 2 years ago
- Non-stationary spectral mixture kernels implemented in GPflow☆28Updated 6 years ago
- ☆22Updated 11 months ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆88Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- ☆80Updated 4 years ago