Didactic multi-output Gaussian processes in Jax.
Project description
MOGPJax
Multi-output Gaussian processes in Jax.
Through extending the GPJax package, MOGPJax aims to provide a low-level interface to multi-output Gaussian process (GP) models in Jax, structured to give researchers maximum flexibility in extending the code to suit their own needs.
Installation
Stable version
To install the latest stable version of MOGPJax run
pip install mogpjax
Development version
To install the latest, possibly unstable, version, the following steps should be followed. It is by no means compulsory, but we do advise that you do all of the below inside a virtual environment.
git clone https://github.com/daniel-dodd/MOGPJax.git
cd MOGPJax
python setup.py develop
We then recommend you check your installation using the supplied unit tests.
python -m pytest tests/
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
MOGPJax-0.0.1.tar.gz
(2.9 kB
view hashes)