pyro-ppl/pyro

Deep universal probabilistic programming with Python and PyTorch

What it solves

Pyro 提供了一种构建和扩展深度概率模型的方式。它允许用户表示任何可计算的概率分布,使得处理不确定性和复杂数据模式更加容易,并且能够在大型数据集上扩展,而不会产生手写代码通常带来的额外开销。

How it works

Pyro 基于 PyTorch,是一种深度概率编程语言(PPL)。它使用一小套强大且可组合的抽象核心,让用户能够表达生成模型和推断模型。它在自动化与手动控制之间取得平衡,为需要自动化的用户提供高级抽象,同时为需要自定义推断过程的专家提供直接访问。

Who it’s for

它面向需要创建灵活、可扩展概率模型的研究人员和开发者,以及那些在高级自动化与低级推断控制之间需要平衡的使用者。

Highlights

  • Universal: 能够表示任何可计算的概率分布。
  • Scalable: 设计用于以最小开销处理大型数据集。
  • Minimal: 采用少量可组合抽象为核心,便于维护。
  • Flexible: 同时提供高级自动化和专家级的推断定制功能。