PyStan或PyMC3是否可以利用机器学习台式机？

问题描述

如果我想最大程度地利用4 GPU机器学习台式机的计算能力，我应该使用pystan或PyMC3进行Python概率编程吗？

解决方法

老实说，GPU对两者的支持都比较差。两者都在进行中。我会去别的地方。

如果您确实要创建性能优化的模型，我的建议是使用TensorFlow probability，因为它具有一流的GPU支持并可以对执行进行相当精细的控制。

还有NumPyro，我没有亲自使用过，但是看起来很有前途。