Quickstart | Install guide | Documentation | Slack Community
GPJax aims to provide a low-level interface to Gaussian process (GP) models in Jax, structured to give researchers maximum flexibility in extending the code to suit their own needs. The idea is that the code should be as close as possible to the maths we write on paper when working with GP models.
Package organisation
Contributions
We would be delighted to receive contributions from interested individuals and groups. To learn how you can get involved, please read our guide for contributing. If you have any questions, we encourage you to open an issue. For broader conversations, such as best GP fitting practices or questions about the mathematics of GPs, we invite you to open a discussion.
Another way you can contribute to GPJax is through issue triaging. This can include reproducing bug reports, asking for vital information such as version numbers and reproduction instructions, or identifying stale issues. If you would like to begin triaging issues, an easy way to get started is to subscribe to GPJax on CodeTriage.
As a contributor to GPJax, you are expected to abide by our code of conduct. If you feel that you have either experienced or witnessed behaviour that violates this standard, then we ask that you report any such behaviours through this form or reach out to one of the project's gardeners.
Feel free to join our Slack Channel, where we can discuss the development of GPJax and broader support for Gaussian process modelling.
We appreciate all the contributors to GPJax who have helped to shape GPJax into the package it is today.
Supported methods and interfaces
Notebook examples
- Conjugate Inference
- Classification
- Sparse Variational Inference
- Stochastic Variational Inference
- Laplace Approximation
- Inference on Non-Euclidean Spaces
- Inference on Graphs
- Heteroscedastic Inference
- Learning Gaussian Process Barycentres
- Deep Kernel Regression
- Poisson Regression
- Bayesian Optimisation
Guides for customisation
Conversion between .ipynb and .py
Above examples are stored in examples directory in the double
percent (py:percent) format. Checkout jupytext
using-cli for more
info.
- To convert
example.pytoexample.ipynb, run:
jupytext --to notebook example.py
- To convert
example.ipynbtoexample.py, run:
jupytext --to py:percent example.ipynb
Installation
Stable version
The latest stable version of GPJax can be installed from PyPI:
pip install gpjax
or from conda-forge:
# with Pixi
pixi add gpjax
# or with conda
conda install --channel conda-forge gpjax
Note
We recommend you check your installation version:
python -c 'import gpjax; print(gpjax.__version__)'
Development version
Warning
This version is possibly unstable and may contain bugs.
Note
We advise you create virtual environment before installing:
conda create -n gpjax_experimental python=3.11.0 conda activate gpjax_experimental
Clone a copy of the repository to your local machine and run the setup configuration in development mode.
git clone https://github.com/thomaspinder/GPJax.git
cd GPJax
uv venv
uv sync --extra dev
We recommend you check your installation passes the supplied unit tests:
uv run poe all-tests
Citing GPJax
If you use GPJax in your research, please cite our JOSS paper.
@article{Pinder2022,
doi = {10.21105/joss.04455},
url = {https://doi.org/10.21105/joss.04455},
year = {2022},
publisher = {The Open Journal},
volume = {7},
number = {75},
pages = {4455},
author = {Thomas Pinder and Daniel Dodd},
title = {GPJax: A Gaussian Process Framework in JAX},
journal = {Journal of Open Source Software}
}
1 maintainer
Release files for gpjax 0.18.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| gpjax-0.18.0.tar.gz | 5.0 MB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| gpjax-0.18.0-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 5.1 MB
Release files / gpjax-0.18.0.tar.gz
| Download URL | gpjax-0.18.0.tar.gz |
|---|---|
| Size | 5.0 MB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
f5900ef75ea07d62dd7af02aaa6c6e29c4c20dcf6962ca12d6168eccf0c88ad2
|
|
BLAKE2b-256 checksum How to use checksums |
7f4094a5545c9cbd690465af8e21a7a08fe6dba996e3511dc77e9f5afbf84f23
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/6.1.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Jul 26, 2026.
Transparency logRelease files / gpjax-0.18.0-py3-none-any.whl
| Download URL | gpjax-0.18.0-py3-none-any.whl |
|---|---|
| Size | 140.3 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
d78489bab734f70b1193d61c7f93ab28fa9f83e3c9f81b6255f4eff3f6a428ae
|
|
BLAKE2b-256 checksum How to use checksums |
007fad95ed293b906ef0c6a1db2e56b3a50be2d263bf551c1be58cda52225cc3
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/6.1.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Jul 26, 2026.
Transparency log
