python - PyStan 或 PyMC3 来利用机器学习桌面？

Question

如果我想最大限度地利用我的 4 GPU 机器学习桌面的计算能力，我应该使用 PyStan 还是 PyMC3 在 Python 中进行概率编程？

Answer 1

老实说，对两者的 GPU 支持都相当粗制滥造。两者都在进行中。我会去别处看看。

如果您真的想创建性能优化模型，我的建议是使用TensorFlow 概率，因为它具有一流的 GPU 支持，并且对执行提供了相当细粒度的控制。

还有NumPyro，我个人没有使用过，但看起来很有希望。