tinygp is built on top of JAX so that’s the primary dependency that you’ll need. All of the methods below will install any required dependencies, but if you want to take advantage of your GPU, that might take a little more setup. tinygp doesn’t have any GPU-specific code, so it should be enough to just follow the installation instructions for CUDA support in the JAX README.

Using pip

The easiest way to install the most recent stable version of tinygp is with pip:

python -m pip install tinygp

From source

Alternatively, you can get the source:

git clone
cd tinygp
python -m pip install -e .


If you installed from source, you can run the unit tests. From the root of the source directory, run:

python -m pip install ".[test]"
python -m pytest -v tests