使用由JAX提供支持的NumPy进行概率编程,可将自动毕业和JIT编译到GPU / TPU / CPU。-Python开发

时间:2024-06-14 09:37:16
【文件属性】:

文件名称:使用由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被设计为轻巧且方便


网友评论