×

NumPyro编号

swMATH ID: 42868
软件作者: 杜凡、尼拉杰·普拉丹、马丁·扬科维亚克
描述: NumPyro中灵活和加速概率规划的合成效果。NumPyro是一个轻量级库,它为Pyro概率编程语言提供了另一个NumPy后端,具有相同的建模接口、语言原语和效果处理抽象。Effect处理程序允许将Pyro的建模API扩展到NumPyro,尽管它构建在一个完全不同的基于JAX的功能后端之上。在这项工作中,我们演示了用程序转换组合Pyro的效果处理程序的强大功能,这些程序转换支持JAX中的硬件加速、自动区分和矢量化。特别是,NumPyro提供了一种可进行端到端JIT编译的无旋转取样器(NUTS)的迭代公式,其实现速度比现有的小型和大型数据集方案快得多。
主页: https://num.pyro.ai/en/stable/
源代码: https://github.com/pyro-ppl/numpyro
关键词: 机器学习;arXiv_状态ML;人工智能;arXiv_cs.AI;arXiv_cs.LG公司;arXiv_cs.PL公司;NumPyro编号;数字Py;烟火;概率规划语言
相关软件: 日本宇宙航空公司;PyMC公司;TensorFlow公司;UCI-毫升;皮斯坦;爱德华二世;XLA公司;烟火;数字Py;GPstuff公司;通用公共管理语言;G流量;GPy公司;GPy火炬;微型GP;高斯过程.jl;PILCO公司;斯坦;斯特诺;黑色JAX
引用于: 1出版物

标准条款

1出版物描述软件 年份
NumPyro中灵活加速概率规划的组合效应
杜凡、尼拉杰·普拉丹、马丁·扬科维亚克
2019

按年份列出的引文