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 4 years ago
- A lightweight didactic library of kernel methods using the back-end JAX.☆12Mar 8, 2023Updated 3 years ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Mar 31, 2021Updated 5 years ago
- Non-stationary spectral mixture kernels implemented in GPflow☆29Nov 30, 2018Updated 7 years ago
- ☆27Aug 27, 2025Updated 11 months 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.
- MBRL library in JAX☆10Sep 22, 2022Updated 3 years ago
- Fully nonstationary, heteroscedastic GP for Matlab☆16Jul 25, 2024Updated 2 years ago
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆243Dec 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
- Extension of almaBTE's Monte Carlo code beyond the relaxation-time approximation.☆10Sep 24, 2023Updated 2 years ago
- AI Agent and CLI for Code Translation and Software Development in Scientific Computing☆18Updated this week
- Implementation for Non-stationary Spectral Kernels (NIPS 2017)☆20Jan 4, 2020Updated 6 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.
- Nonparametric Differential Equation Modeling☆56Mar 5, 2024Updated 2 years ago
- Learning unknown ODE models with Gaussian processes☆28Jul 30, 2018Updated 8 years ago
- Tutorials and sampling algorithm comparisons☆86Updated this week
- Hands-On Reinforcement Learning with TensorFlow & TRFL☆14Jan 18, 2021Updated 5 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
- A Simple Statistical Distribution Library in JAX☆16Mar 30, 2024Updated 2 years ago
- Code of PriorVAE: encoding spatial priors with variational autoencoders☆12Jul 13, 2023Updated 3 years ago
- Deep GPs built on top of TensorFlow/Keras and GPflow☆128Oct 16, 2024Updated last year
- Implementation of GPLVM and Bayesian GPLVM in pytorch/gpytorch☆15Jun 21, 2021Updated 5 years ago
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- Performing parameter estimation on gravitational wave data with machine learning☆21May 4, 2026Updated 2 months ago
- ☆20Dec 8, 2022Updated 3 years ago
- Lecture material used in my 8 hour long PhD course on "Numerical Methods for Partial Differential Equations"☆16May 21, 2025Updated last year
- Code for efficiently sampling functions from GP(flow) posteriors☆75Nov 13, 2020Updated 5 years ago
- An HMC/NUTS implementation in Aesara☆31Jun 22, 2023Updated 3 years ago
- Gaussian processes in JAX and Equinox.☆643Updated this week
- Model-based reinforcement learning in TensorFlow☆57Jul 27, 2021Updated 5 years ago
- ☆16Jan 19, 2024Updated 2 years ago
- ☆12Jul 6, 2022Updated 4 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- A Bayesian optimization toolbox built on TensorFlow☆251Jun 17, 2026Updated last month
- Hands-on Reinforcement Learning with TensorFlow by Packt Publishing☆13Jan 15, 2021Updated 5 years ago
- Implementation of Neurips 2023 Paper "Multi Time Scale World Models"☆18Nov 8, 2024Updated last year
- HMC and other derivative-based MCMC sampling algorithms for the nimble package☆13Jul 20, 2026Updated 2 weeks ago
- Methods and experiments for assumed density SDE approximations☆12Jan 26, 2022Updated 4 years ago
- MLflow logging for PyMC☆14Aug 31, 2025Updated 11 months ago
- A code to simulate linearized phonon transport☆15May 7, 2017Updated 9 years ago