pith. sign in

arxiv: 1907.01063 · v2 · pith:B4QSI6EVnew · submitted 2019-07-01 · 💻 cs.MS · cs.DC· stat.CO

GPU-based Parallel Computation Support for Stan

classification 💻 cs.MS cs.DCstat.CO
keywords stanregressionadditionsalgebraallowsbenefitchangescholesky
0
0 comments X
read the original abstract

This paper details an extensible OpenCL framework that allows Stan to utilize heterogeneous compute devices. It includes GPU-optimized routines for the Cholesky decomposition, its derivative, other matrix algebra primitives and some commonly used likelihoods, with more additions planned for the near future. Stan users can now benefit from large speedups offered by GPUs with little effort and without changes to their existing Stan code. We demonstrate the practical utility of our work with two examples - logistic regression and Gaussian Process regression.

This paper has not been read by Pith yet.

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.