标签: numpyro

来自 Pytorch-Pyro 的网站：

我们很高兴地宣布发布 NumPyro，这是一个 NumPy 支持的 Pyro，使用 JAX 进行自动微分和 JIT 编译，HMC 和 NUTS 的速度提高了 100 倍以上！

我的问题：

1. NumPyro（超过 Pyro）的性能提升（有时是340 倍或 2 倍）究竟来自哪里？
2. 更重要的是，为什么（而不是，在哪里）我会继续使用 Pyro？

额外的：

1. 与 Tensorflow Probability 相比，我应该如何查看 NumPyro 的性能和特性，以决定在何处使用哪个？