Minimal Gaussian process library in JAX with a simple (custom) approach to state management.
☆12Dec 20, 2023Updated 2 years ago
Alternatives and similar repositories for GPJax
Users that are interested in GPJax are comparing it to the libraries listed below
Sorting:
- Mixtures of Gaussian Process Experts in GPflow/TensorFlow☆12Aug 1, 2022Updated 3 years ago
- A lightweight didactic library of kernel methods using the back-end JAX.☆12Mar 8, 2023Updated 2 years ago
- Non-stationary spectral mixture kernels implemented in GPflow☆29Nov 30, 2018Updated 7 years ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Mar 31, 2021Updated 4 years ago
- Fully nonstationary, heteroscedastic GP for Matlab☆16Jul 25, 2024Updated last year
- Bayesian inference for a logistic regression model in various languages☆43Jul 12, 2023Updated 2 years ago
- Fully Bayesian Inference in GPs - Gaussian and Generic Likelihoods☆22Sep 26, 2023Updated 2 years ago
- Implementation for Non-stationary Spectral Kernels (NIPS 2017)☆20Jan 4, 2020Updated 6 years ago
- ☆21Dec 8, 2022Updated 3 years ago
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆241Dec 22, 2023Updated 2 years ago
- Nonparametric Differential Equation Modeling☆56Mar 5, 2024Updated 2 years ago
- An HMC/NUTS implementation in Aesara☆31Jun 22, 2023Updated 2 years ago
- Learning unknown ODE models with Gaussian processes☆27Jul 30, 2018Updated 7 years ago
- Online variational GPs☆38Jun 1, 2023Updated 2 years ago
- Deep GPs built on top of TensorFlow/Keras and GPflow☆128Oct 16, 2024Updated last year
- Tutorials and sampling algorithm comparisons☆80Updated this week
- Gaussian processes in JAX and Flax.☆596Updated this week
- Kinematic and dynamic models of continuum and articulated soft robots.☆15Nov 22, 2025Updated 3 months ago
- Code for efficiently sampling functions from GP(flow) posteriors☆74Nov 13, 2020Updated 5 years ago
- A Bayesian optimization toolbox built on TensorFlow☆248Jan 14, 2026Updated last month
- A python implementation of the COACH algorithm for the Cartpole problem in OpenAI gym.☆11Mar 15, 2019Updated 6 years ago
- ☆14Mar 3, 2022Updated 4 years ago
- Water wave models in one dimension☆10Feb 24, 2026Updated last week
- Curatable database for experimental and theoretical data on solid materials.☆13Sep 21, 2025Updated 5 months ago
- IterGP: Computation-Aware Gaussian Process Inference (NeurIPS 2022)☆43Apr 12, 2023Updated 2 years ago
- Reference implementation of algorithms for reinforcement learning and Markov decision processes.☆12Jan 28, 2021Updated 5 years ago
- Data for Coordination Complexes for the InChI Identifier☆12May 11, 2021Updated 4 years ago
- [ICLR 2020] Learning Compositional Koopman Operators for Model-Based Control☆92Apr 5, 2021Updated 4 years ago
- ☆15Feb 16, 2026Updated 2 weeks ago
- ☆10Mar 25, 2023Updated 2 years ago
- Notebooks accompanying the paper on GPR in materials and molecules in Chemical Reviews 2020☆12May 4, 2021Updated 4 years ago
- ☆10Jul 25, 2022Updated 3 years ago
- The good practice in the VQA system such as pos-tag attention, structed triplet learning and triplet attention is very general and can be…☆19Jan 23, 2018Updated 8 years ago
- A package for the simulation of gas chromatography (GC)☆16Jan 24, 2026Updated last month
- ☆15Dec 31, 2023Updated 2 years ago
- Definitive Screening design of experiments☆13May 28, 2024Updated last year
- Repo for the Journal of Systems Research (JSYS)☆11Jan 7, 2020Updated 6 years ago
- An unofficial jax/haiku implementation of Crystal Graph Convolutional Neural Networks (CGCNN)☆10Dec 17, 2022Updated 3 years ago
- Uncertainty sets for nonlinear dynamical systems☆10Nov 7, 2020Updated 5 years ago