PyStan or PyMC3 to take advantage of machine learning desktop?

1.3k views Asked by At

If I want to take maximum advantage of the computing power of my 4 GPU machine learning desktop, should I use PyStan or PyMC3 for probabilistic programming in Python?

1

There are 1 answers

0
fny On

Honestly, GPU support for both is rather shoddy. Both are works in progress. I'd look elsewhere.

If you're really trying to create a performance optimized model, my recommendation would be to use TensorFlow probability since it has first class GPU support and gives fairly fine-grained control over execution.

There's also NumPyro, which I haven't used personally but looks promising.