JAX 标签的文章

使用 JAX 和 Numba 更快地采样

PyMC 可以通过 PyTensor 将其模型编译到各种执行后端,包括

阅读更多 ...


Pathfinder 变分推断

Pathfinder [Zhang et al., 2021] 是一种变分推断算法,可以从贝叶斯模型的后验分布中生成样本。它比广泛使用的 ADVI 算法更具优势。在大型问题上,与大多数 MCMC 算法(包括动态 HMC(即 NUTS))相比,它应该具有更好的扩展性,但代价是对后验分布的估计更具偏差。有关该算法的详细信息,请参阅 arxiv 预印本

阅读更多 ...


如何在 PyMC 中包装 JAX 函数以供使用

此笔记本使用非 PyMC 依赖库,因此需要专门安装才能运行此笔记本。打开下面的下拉菜单以获取更多指导。

阅读更多 ...