跳转到主要内容

PyTorch中高斯过程的一个实现

项目描述

GPyTorch


Test Suite Documentation Status License

Python Version Conda PyPI

GPyTorch是使用PyTorch实现的高斯过程库。GPyTorch旨在轻松创建可扩展、灵活和模块化的高斯过程模型。

内部,GPyTorch与许多现有的GP推理方法不同,它通过使用数值线性代数技术(如预条件共轭梯度)来执行大多数推理操作。实现可扩展的GP方法就像通过我们的LinearOperator接口提供矩阵乘法例程,或者通过组合我们现有的许多LinearOperators一样简单。这不仅使得流行可扩展的GP技术的实现变得容易,而且与基于Cholesky分解的求解器相比,通常还能显著提高GPU计算的使用率。

PyTorch GPyTorch 提供以下功能:(1) 显著的 GPU 加速(通过基于 MVM 的推理);(2) 最先进的最新算法进步的实现,以实现可扩展性和灵活性(例如,SKI/KISS-GP随机 Lanczos 展开式LOVESKIP随机变分 深度核学习 等);(3) 与深度学习框架的简单集成。

示例、教程和文档

请参阅我们的 文档、示例和教程,了解如何在 GPyTorch 中构建各种模型。

安装

要求:

  • Python >= 3.8
  • PyTorch >= 2.0

使用 pip 或 conda 安装 GPyTorch

pip install gpytorch
conda install gpytorch -c gpytorch

(若要全局使用包但仅将 GPyTorch 安装为用户包,请使用上述 pip install --user 命令。)

最新(不稳定)版本

要升级到最新(不稳定)版本,请运行

pip install --upgrade git+https://github.com/cornellius-gp/linear_operator.git
pip install --upgrade git+https://github.com/cornellius-gp/gpytorch.git

开发版本

如果您正在提交拉取请求,最好进行手动安装

git clone https://github.com/cornellius-gp/gpytorch.git
cd gpytorch
pip install -e .[dev,docs,examples,keops,pyro,test]  # keops and pyro are optional

ArchLinux 软件包

注意:实验性 AUR 软件包。对于大多数用户,我们建议使用 conda 或 pip 进行安装。

GPyTorch 还可在 ArchLinux 用户仓库(AUR)中找到。您可以使用 AUR 辅助工具,如 yay,按照以下方式安装:

yay -S python-gpytorch

要讨论与该 AUR 软件包相关的任何问题,请参阅 python-gpytorch 的评论部分。

引用我们

如果您使用 GPyTorch,请引用以下论文

Gardner, Jacob R.、Geoff Pleiss、David Bindel、Kilian Q. Weinberger 和 Andrew Gordon Wilson。 "GPyTorch:带 GPU 加速的 Blackbox 矩阵-矩阵高斯过程推理。" 发表于神经信息处理系统(2018)。

@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}
}

贡献

有关提交问题和拉取请求的信息,请参阅 CONTRIBUTING.md

团队

GPyTorch 主要由以下人员维护:

我们还要感谢我们的其他贡献者,包括但不限于 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 等!

致谢

GPyTorch 的发展得到了以下机构的资助:比尔及梅琳达·盖茨基金会、美国国家科学基金会、SAP、西蒙斯基金会和盖茨慈善信托。

许可证

GPyTorch 采用 MIT 许可证

项目详情


下载文件

下载适合您平台的文件。如果您不确定选择哪个,请了解有关 安装包 的更多信息。

源代码分发

gpytorch-1.13.tar.gz (2.5 MB 查看哈希值)

上传时间 源代码

构建分发

gpytorch-1.13-py3-none-any.whl (277.8 kB 查看哈希值)

上传时间 Python 3

由以下机构支持