GPyTorch

logo

A highly efficient and modular implementation of GPs, with GPU acceleration.Implemented in PyTorch.

Examples

Browse Examples

Documentation

Browse Docs

To learn about GPyTorch's inference engine, please refer to our NeurIPS 2018 paper:
GPyTorch: Blackbox Matrix-Matrix Gaussian Process Inference with GPU Acceleration

ArXiVBibTeX

Installation

Make sure you have PyTorch installed. Then,

pip install gpytorch

For more instructions, see the Github README.

The Team

With (many) contributions from:  Eytan Bakshy, Wesley Maddox, Ke Alexander Wang, Ruihan Wu, Sait Cakmak, David Eriksson, Sam Daulton, Martin Jankowiak, Sam Stanton, Zitong Zhou, David Arbour, Karthik Rajkumar, Bram Wallace, Jared Frank, and others!

Developed at Cornell University, Columbia University, University of Pennsylvania, New York University, and Meta, with funding from the Bill and Melinda Gates Foundation, the National Science Foundation,SAP, the Simons Foundation, and the Gatsby Charatable Trust.