google-research / gpax
☆24Updated this week
Related projects: ⓘ
- Bayesian algorithm execution (BAX)☆46Updated 3 years ago
- Bayesian inference with Python and Jax.☆31Updated last year
- Pre-trained Gaussian processes for Bayesian optimization☆78Updated 2 weeks ago
- Riemannian Optimization Using JAX☆44Updated 10 months ago
- Normalizing Flows using JAX☆82Updated 9 months ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆34Updated 3 years ago
- ☆33Updated 2 years ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)☆20Updated 3 years ago
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆11Updated 9 months ago
- ☆78Updated 3 years ago
- Turning SymPy expressions into JAX functions☆42Updated 3 years ago
- Implicit Deep Adaptive Design (iDAD): Policy-Based Experimental Design without Likelihoods☆17Updated 2 years ago
- Riemannian Convex Potential Maps☆68Updated last year
- [ICLR 2022] Path integral sampler☆40Updated last year
- Public code for running Stochastic Gradient Descent on GPs.☆31Updated 5 months ago
- Inference Combinators in JAX☆42Updated 2 months ago
- Multiple dispatch over abstract array types in JAX.☆100Updated last month
- ☆49Updated last year
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆94Updated 10 months ago
- Code for efficiently sampling functions from GP(flow) posteriors☆65Updated 3 years ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated last year
- A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes☆25Updated 2 years ago
- Turn jitted jax functions back into python source code☆20Updated 2 months ago
- simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…☆14Updated 3 years ago
- Loopy belief propagation for factor graphs on discrete variables in JAX☆124Updated 3 months ago
- Automatic Functional Differentiation in JAX☆51Updated 7 months ago
- ☆30Updated 2 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆82Updated 4 months ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆35Updated last year