google-research / gpax
☆30Updated 6 months ago
Alternatives and similar repositories for gpax:
Users that are interested in gpax are comparing it to the libraries listed below
- Pre-trained Gaussian processes for Bayesian optimization☆90Updated 3 months ago
- Bayesian inference with Python and Jax.☆32Updated 2 years ago
- Riemannian Optimization Using JAX☆48Updated last year
- Code for efficiently sampling functions from GP(flow) posteriors☆70Updated 4 years ago
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆11Updated last year
- Normalizing Flows using JAX☆83Updated last year
- ☆80Updated 3 years ago
- Matrix-free linear algebra in JAX.☆116Updated 2 months ago
- ☆49Updated 2 years ago
- Turn jitted jax functions back into python source code☆22Updated 3 months ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆42Updated last year
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- Implicit Deep Adaptive Design (iDAD): Policy-Based Experimental Design without Likelihoods☆19Updated 3 years ago
- A generic interface for linear algebra backends☆71Updated 3 weeks ago
- Public code for running Stochastic Gradient Descent on GPs.☆36Updated 4 months ago
- Riemannian Convex Potential Maps☆67Updated last year
- Laplace Redux -- Effortless Bayesian Deep Learning☆42Updated last year
- ☆30Updated 2 years ago
- Turning SymPy expressions into JAX functions☆44Updated 4 years ago
- Code for the paper "Randomly pivoted Cholesky: Practical approximation of a kernel matrix with few entry evaluations"☆28Updated 5 months ago
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆102Updated last year
- ☆34Updated 3 years ago
- Bayesian Optimization with Density-Ratio Estimation☆23Updated 2 years ago
- Neural likelihood-free methods in PyTorch.☆39Updated 5 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Bayesian Neural Network Surrogates for Bayesian Optimization☆48Updated 10 months ago
- [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)☆21Updated 4 years ago
- Multiple dispatch over abstract array types in JAX.☆115Updated 3 weeks ago
- Code associated with paper "High-Dimensional Contextual Policy Search with Unknown Context Rewards using Bayesian Optimization"☆15Updated 4 years ago