文件名称:使用由JAX提供支持的NumPy进行概率编程,可将自动毕业和JIT编译到GPU / TPU / CPU。-Python开发
文件大小:3.14MB
文件格式:ZIP
更新时间:2024-06-14 09:37:16
Python Deep Learning
NumPyro由JAX提供支持的NumPy进行概率编程,可自动将其和JIT编译到GPU / TPU / CPU。 文件| 例子 论坛什么是NumPyro? NumPyro是一个小型的概率编程库,它使用由JAX驱动的NumPy进行NumPyro概率编程,可将Autograd和JIT编译到GPU / TPU / CPU。 文件| 例子 论坛什么是NumPyro? NumPyro是一个小型的概率编程库,为Pyro提供了NumPy后端。 我们依靠JAX进行自动区分,并将JIT编译到GPU / CPU。 这是一个处于积极开发阶段的Alpha版本,因此请当心设计的脆弱性,错误和对API的更改。 NumPyro被设计为轻巧且方便