Dec, 2019

可组合的效应:NumPyro 中灵活而加速的概率编程

TL;DR本文介绍了NumPyro库,它是Pyro概率编程语言的一个轻量级后端,具有相同的建模接口、语言原语和效果处理抽象。效果处理程序允许将Pyro的建模API扩展到NumPyro,本文演示了使用程序变换与Pyro的效果处理程序组合的强大功能。所提供的迭代式No-U-Turn采样器(NUTS)的NumPyro实现,具有比现有替代方案更快的速度,适用于小型和大型数据集。