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
- MBRL library in JAX☆10Sep 22, 2022Updated 3 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- 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☆15Jan 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
- Learning unknown ODE models with Gaussian processes☆27Jul 30, 2018Updated 7 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 5 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.
- ☆27Aug 27, 2025Updated 9 months ago
- Deep GPs built on top of TensorFlow/Keras and GPflow☆129Oct 16, 2024Updated last year
- An HMC/NUTS implementation in Aesara☆31Jun 22, 2023Updated 2 years ago
- ☆20Dec 8, 2022Updated 3 years ago
- Model-based reinforcement learning in TensorFlow☆57Jul 27, 2021Updated 4 years ago
- Code for efficiently sampling functions from GP(flow) posteriors☆75Nov 13, 2020Updated 5 years ago
- Gaussian processes in JAX and Equinox.☆628Jun 15, 2026Updated last week
- ☆12Jul 6, 2022Updated 3 years ago
- Methods and experiments for assumed density SDE approximations☆12Jan 26, 2022Updated 4 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Implementation of Neurips 2023 Paper "Multi Time Scale World Models"☆17Nov 8, 2024Updated last year
- A Simple Statistical Distribution Library in JAX☆16Mar 30, 2024Updated 2 years ago
- MLflow logging for PyMC☆14Aug 31, 2025Updated 9 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
- Cookiecutter template for an R analysis project☆18Apr 4, 2018Updated 8 years ago
- Kinematic and dynamic models of continuum and articulated soft robots.☆16Nov 22, 2025Updated 6 months ago
- Basic Gaussian process regression library. (Eigen3 required)☆27Jul 31, 2015Updated 10 years ago
- [ICLR 2020] Learning Compositional Koopman Operators for Model-Based Control☆94Apr 5, 2021Updated 5 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.
- Bayesian spatio-temporal methods for small-area estimation of HIV indicators (PhD thesis, Imperial College London, 2023)☆16Nov 11, 2025Updated 7 months ago
- State-space model inference with JAX☆67Updated this week
- The official implementation of Non-separable Spatio-temporal Graph Kernels via SPDEs.☆16Jun 2, 2022Updated 4 years ago
- ☆40Nov 11, 2021Updated 4 years ago
- Simple Xarray + JAX Integration☆21Mar 21, 2026Updated 3 months ago
- Code and submissions for the 30 Day Map Challenge in November 2021☆15Nov 26, 2021Updated 4 years ago
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆34Feb 10, 2026Updated 4 months ago