google-research / gpax
☆30Updated 5 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
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆11Updated last year
- Code for efficiently sampling functions from GP(flow) posteriors☆70Updated 4 years ago
- ☆49Updated 2 years ago
- Turn jitted jax functions back into python source code☆22Updated 2 months ago
- [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)☆21Updated 4 years ago
- Riemannian Optimization Using JAX☆48Updated last year
- ☆80Updated 3 years ago
- Minimal Implementation of Bayesian Optimization in JAX☆88Updated 2 months ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆101Updated last year
- A generic interface for linear algebra backends☆71Updated last week
- Implicit Deep Adaptive Design (iDAD): Policy-Based Experimental Design without Likelihoods☆19Updated 3 years ago
- Normalizing Flows using JAX☆82Updated last year
- Code associated with paper "High-Dimensional Contextual Policy Search with Unknown Context Rewards using Bayesian Optimization"☆15Updated 4 years ago
- ☆15Updated 2 years ago
- Bayesian Optimization over Discrete and Mixed Spaces via Probabilistic Reparameterization☆72Updated 2 years ago
- Graph Learning with JAX☆14Updated 2 years ago
- Code for A General Recipe for Likelihood-free Bayesian Optimization, ICML 2022☆44Updated 2 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆25Updated last month
- simple JAX-/NumPy-based implementations of NGD with exact/approximate Fisher Information Matrix both in parameter-space and function-spac…☆14Updated 4 years ago
- Turning SymPy expressions into JAX functions☆44Updated 3 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆44Updated 2 months ago
- Multiple dispatch over abstract array types in JAX.☆114Updated 2 weeks ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- Bayesian Optimization with Density-Ratio Estimation☆23Updated 2 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago