Installation

Stable version

The latest stable release 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

Check your installation

We recommend you check your installation version:

python -c 'import gpjax; print(gpjax.__version__)'

GPU/TPU support

Fancy using GPJax on GPU/TPU? Then you'll need to install JAX with the relevant hardware acceleration support as detailed in the JAX installation guide.

Development version

Warning

This version is possibly unstable and may contain bugs.

The latest development version of GPJax can be installed via running following:

git clone https://github.com/thomaspinder/GPJax.git
cd GPJax
uv sync --extra dev

Tip

We advise you create virtual environment before installing:

conda create -n gpjax_experimental python=3.10.0
conda activate gpjax_experimental

and recommend you check your installation passes the supplied unit tests:

uv run poe all-tests