Installation

Stable version

The latest stable release of GPJax can be installed via pip:

pip install 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
poetry install

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:

poetry run pytest tests/