An implementation of Gaussian Processes in Pytorch
Project description
GPyTorch
News: GPyTorch v1.0.0
GPyTorch v1.0.0 has just been released. This release marks our exit from beta status and in to what we consider stable software. This means that we do not expect you to encounter any major bugs when using stable features. Check out the release notes, as well as our fully revamped documentation and example notebooks.
GPyTorch is a Gaussian process library implemented using PyTorch. GPyTorch is designed for creating scalable, flexible, and modular Gaussian process models with ease.
Internally, GPyTorch differs from many existing approaches to GP inference by performing all inference operations using modern numerical linear algebra techniques like preconditioned conjugate gradients. Implementing a scalable GP method is as simple as providing a matrix multiplication routine with the kernel matrix and its derivative via our LazyTensor interface, or by composing many of our already existing LazyTensors. This allows not only for easy implementation of popular scalable GP techniques, but often also for significantly improved utilization of GPU computing compared to solvers based on the Cholesky decomposition.
GPyTorch provides (1) significant GPU acceleration (through MVM based inference); (2) state-of-the-art implementations of the latest algorithmic advances for scalability and flexibility (SKI/KISS-GP, stochastic Lanczos expansions, LOVE, SKIP, stochastic variational deep kernel learning, ...); (3) easy integration with deep learning frameworks.
Examples, Tutorials, and Documentation
See our numerous examples and tutorials on how to construct all sorts of models in GPyTorch.
Installation
Requirements:
- Python >= 3.6
- PyTorch >= 1.3
Install GPyTorch using pip or conda:
pip install gpytorch
conda install gpytorch -c gpytorch
(To use packages globally but install GPyTorch as a user-only package, use pip install --user above.)
Latest (unstable) version
To upgrade to the latest (unstable) version, run
pip install --upgrade git+https://github.com/cornellius-gp/gpytorch.git
Citing Us
If you use GPyTorch, please cite the following papers:
@inproceedings{gardner2018gpytorch,
title={GPyTorch: Blackbox Matrix-Matrix Gaussian Process Inference with GPU Acceleration},
author={Gardner, Jacob R and Pleiss, Geoff and Bindel, David and Weinberger, Kilian Q and Wilson, Andrew Gordon},
booktitle={Advances in Neural Information Processing Systems},
year={2018}
}
Development
To run the unit tests:
python -m unittest
By default, the random seeds are locked down for some of the tests. If you want to run the tests without locking down the seed, run
UNLOCK_SEED=true python -m unittest
If you plan on submitting a pull request, please make use of our pre-commit hooks to ensure that your commits adhere to the general style guidelines enforced by the repo. To do this, navigate to your local repository and run:
pip install pre-commit
pre-commit install
From then on, this will automatically run flake8, isort, black and other tools over the files you commit each time you commit to gpytorch or a fork of it.
The Team
GPyTorch is primarily maintained by:
- Jake Gardner (Uber AI Labs)
- Geoff Pleiss (Cornell University)
- Kilian Weinberger (Cornell University)
- Andrew Gordon Wilson (Cornell University)
- Max Balandat (Facebook)
We would like to thank our other contributors including (but not limited to) David Arbour, Eytan Bakshy, David Eriksson, Jared Frank, Sam Stanton, Bram Wallace, Ke Alexander Wang, Ruihan Wu.
Acknowledgements
Development of GPyTorch is supported by funding from the Bill and Melinda Gates Foundation, the National Science Foundation, and SAP.
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
File details
Details for the file gpytorch-1.0.1.tar.gz.
File metadata
- Download URL: gpytorch-1.0.1.tar.gz
- Upload date:
- Size: 229.6 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/1.13.0 pkginfo/1.4.2 requests/2.21.0 setuptools/40.0.0 requests-toolbelt/0.9.1 tqdm/4.37.0 CPython/3.6.8
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 3c18464f77c463d1013ad8ad897bf1658fb5c798ed69ec050537771b1aeb5c74 |
|
| MD5 | 3acfb6f01d780f655ddfb5145588ff78 |
|
| BLAKE2b-256 | 9c5fce79e35c1a36deb25a0eac0f67bfe85fb8350eb8e19223950c3d615e5e9a |
