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☆12Aug 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
- Non-stationary spectral mixture kernels implemented in GPflow☆29Nov 30, 2018Updated 7 years ago
- Numerical relativity surrogate waveform in Jax☆19Aug 14, 2025Updated 8 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- MBRL library in JAX☆11Sep 22, 2022Updated 3 years ago
- Fully nonstationary, heteroscedastic GP for Matlab☆16Jul 25, 2024Updated last year
- 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 2 years ago
- Decorators for logging purposes for all your dataframes☆14Jan 31, 2025Updated last year
- Fully Bayesian Inference in GPs - Gaussian and Generic Likelihoods☆22Sep 26, 2023Updated 2 years ago
- Code and lecture materials for the "Data Science for Physicists" short course held at the 2025 Joint March and April Meetings☆21Mar 31, 2025Updated last year
- Nonparametric Differential Equation Modeling☆56Mar 5, 2024Updated 2 years ago
- Implementation for Non-stationary Spectral Kernels (NIPS 2017)☆20Jan 4, 2020Updated 6 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Learning unknown ODE models with Gaussian processes☆27Jul 30, 2018Updated 7 years ago
- Performing parameter estimation on gravitational wave data with machine learning☆21Updated this week
- Tutorials and sampling algorithm comparisons☆82Apr 14, 2026Updated last week
- Hands-On Reinforcement Learning with TensorFlow & TRFL☆14Jan 18, 2021Updated 5 years ago
- ☆16Jan 19, 2024Updated 2 years ago
- Extension of almaBTE's Monte Carlo code beyond the relaxation-time approximation.☆10Sep 24, 2023Updated 2 years ago
- Implementation of GPLVM and Bayesian GPLVM in pytorch/gpytorch☆15Jun 21, 2021Updated 4 years ago
- ☆27Aug 27, 2025Updated 7 months ago
- Deep GPs built on top of TensorFlow/Keras and GPflow☆128Oct 16, 2024Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- An HMC/NUTS implementation in Aesara☆31Jun 22, 2023Updated 2 years ago
- ☆21Dec 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☆56Jul 27, 2021Updated 4 years ago
- Gaussian processes in JAX and Equinox.☆612Updated this week
- ☆12Jul 6, 2022Updated 3 years ago
- Hands-on Reinforcement Learning with TensorFlow by Packt Publishing☆13Jan 15, 2021Updated 5 years ago
- A Bayesian optimization toolbox built on TensorFlow☆248Updated this week
- HMC and other derivative-based MCMC sampling algorithms for the nimble package☆13Dec 15, 2025Updated 4 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Methods and experiments for assumed density SDE approximations☆12Jan 26, 2022Updated 4 years ago
- Implementation of Neurips 2023 Paper "Multi Time Scale World Models"☆16Nov 8, 2024Updated last year
- measure minimum solid/void lengthscales in binary image☆22Apr 1, 2026Updated 2 weeks ago
- A Simple Statistical Distribution Library in JAX☆16Mar 30, 2024Updated 2 years ago
- MLflow logging for PyMC☆14Aug 31, 2025Updated 7 months ago
- Implementation of the work Variational multiple shooting for Bayesian ODEs with Gaussian processes☆14Aug 5, 2022Updated 3 years ago
- Code of PriorVAE: encoding spatial priors with variational autoencoders☆12Jul 13, 2023Updated 2 years ago