文件名称：numpyro:使用由JAX提供支持的NumPy进行概率编程，可将Autograd和JIT编译为GPUTPUCPU
文件大小：3.13MB
文件格式：ZIP
更新时间：2024-05-31 11:31:11
numpy probabilistic-programming bayesian-inference hmc pyro
NumPyro 使用由提供支持的NumPy进行概率编程，可将自动毕业和JIT编译到GPU / TPU / CPU。 | 什么是NumPyro？ NumPyro是一个小型的概率编程库，为提供了NumPy后端。 我们依靠进行自动区分，并将JIT编译到GPU / CPU。 这是一个处于积极开发阶段的Alpha版本，因此请当心设计的脆弱性，错误和对API的更改。 NumPyro的设计轻巧，致力于提供用户可在其上构建的灵活基材： Pyro基本体： NumPyro程序可以包含常规Python和NumPy代码，此外还有sample和param类的。 除了PyTorch和Numpy的API之间的一些细微差别外，模型代码看起来应该与Pyro非常相似。 请参见下面的。 推理算法： NumPyro当前支持汉密尔顿蒙特卡洛，包括No U-Turn采样器的实现。 NumPyro的动机之一是通过JIT编译