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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Mixtures of Gaussian Process Experts in GPflow/TensorFlow☆13Aug 1, 2022Updated 3 years ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Mar 31, 2021Updated 5 years ago
- A lightweight didactic library of kernel methods using the back-end JAX.☆12Mar 8, 2023Updated 3 years ago
- Numerical relativity surrogate waveform in Jax☆19Aug 14, 2025Updated 10 months ago
- MBRL library in JAX☆10Sep 22, 2022Updated 3 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆242Dec 22, 2023Updated 2 years ago
- Bayesian inference for a logistic regression model in various languages☆43Jul 12, 2023Updated 3 years ago
- Decorators for logging purposes for all your dataframes☆15Jan 31, 2025Updated last year
- Fully Bayesian Inference in GPs - Gaussian and Generic Likelihoods☆22Sep 26, 2023Updated 2 years ago
- Nonparametric Differential Equation Modeling☆56Mar 5, 2024Updated 2 years ago
- Implementation for Non-stationary Spectral Kernels (NIPS 2017)☆20Jan 4, 2020Updated 6 years ago
- Learning unknown ODE models with Gaussian processes☆28Jul 30, 2018Updated 7 years ago
- Performing parameter estimation on gravitational wave data with machine learning☆21May 4, 2026Updated 2 months ago
- Hands-On Reinforcement Learning with TensorFlow & TRFL☆14Jan 18, 2021Updated 5 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- ☆16Jan 19, 2024Updated 2 years ago
- Implementation of GPLVM and Bayesian GPLVM in pytorch/gpytorch☆15Jun 21, 2021Updated 5 years ago
- ☆27Aug 27, 2025Updated 10 months ago
- Deep GPs built on top of TensorFlow/Keras and GPflow☆128Oct 16, 2024Updated last year
- An HMC/NUTS implementation in Aesara☆31Jun 22, 2023Updated 3 years ago
- ☆20Dec 8, 2022Updated 3 years ago
- Code for efficiently sampling functions from GP(flow) posteriors☆75Nov 13, 2020Updated 5 years ago
- Model-based reinforcement learning in TensorFlow☆57Jul 27, 2021Updated 4 years ago
- Gaussian processes in JAX and Equinox.☆637Jul 6, 2026Updated last week
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Hands-on Reinforcement Learning with TensorFlow by Packt Publishing☆13Jan 15, 2021Updated 5 years ago
- A Bayesian optimization toolbox built on TensorFlow☆251Jun 17, 2026Updated 3 weeks ago
- HMC and other derivative-based MCMC sampling algorithms for the nimble package☆13Jul 2, 2026Updated last week
- Methods and experiments for assumed density SDE approximations☆12Jan 26, 2022Updated 4 years ago
- measure minimum solid/void lengthscales in binary image☆22Jul 1, 2026Updated last week
- A Simple Statistical Distribution Library in JAX☆16Mar 30, 2024Updated 2 years ago
- Implementation of the work Variational multiple shooting for Bayesian ODEs with Gaussian processes☆13Aug 5, 2022Updated 3 years ago
- Code of PriorVAE: encoding spatial priors with variational autoencoders☆12Jul 13, 2023Updated 3 years ago
- Cookiecutter template for an R analysis project☆18Apr 4, 2018Updated 8 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- This repository contains the machine learning projects completed for the class "Deep Learning in Scientific Computing" taught at ETH join…☆25Apr 7, 2024Updated 2 years ago
- Kinematic and dynamic models of continuum and articulated soft robots.☆16Nov 22, 2025Updated 7 months ago
- Basic Gaussian process regression library. (Eigen3 required)☆28Jul 31, 2015Updated 10 years ago
- Bayesian spatio-temporal methods for small-area estimation of HIV indicators (PhD thesis, Imperial College London, 2023)☆16Nov 11, 2025Updated 8 months ago
- 📖 Has the amount of text on Vogue magazines changed?☆10Sep 8, 2024Updated last year
- State-space model inference with JAX☆71Jul 4, 2026Updated last week
- The official implementation of Non-separable Spatio-temporal Graph Kernels via SPDEs.☆16Jun 2, 2022Updated 4 years ago