GPflow/GPflow

Gaussian processes in TensorFlow

What it solves

GPflow 是一个 Python 包,旨在简化高斯过程(GP)模型的构建。它提供了一个框架,用于实现现代高斯过程推断,支持可组合的 kernel 与 likelihood,以处理复杂的数据模式。

How it works

基于 TensorFlow 2.4+ 与 TensorFlow Probability,GPflow 利用这两个库执行底层计算。此架构使模型能够在 GPU 上快速运行,从而更高效地处理更大的数据集。

Who it’s for

它面向需要构建和部署高斯过程模型的研究人员和开发者,应用场景包括回归、贝叶斯优化以及分子属性预测等任务。

Highlights

  • GPU Acceleration: Uses TensorFlow to enable fast execution on GPUs.
  • Composable Components: Supports the creation of complex models through composable kernels and likelihoods.
  • Extensible Framework: Serves as the foundation for other specialized projects like Trieste for Bayesian optimization.
  • Modern Inference: Implements up-to-date Gaussian process inference techniques.