×

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。人工智能arXiv_cs。LG公司arXiv_cs。损益NumPyro公司数字Py烟火概率规划语言
相关软件: 螺母烟火日本宇宙航空公司github亚当PyMC公司TensorFlow公司火炬差异乌兹拉布近五年NIFTY公司DeepONet(深度网络)埃皮埃斯特姆肾上腺素MNIST公司ramcmc公司电影镜头GPstuff公司通用公共管理语言G流量
引用于: 6文件

标准条款

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

按年份列出的引文