Conference site ยป Proceedings

gpustats: GPU Library for Statistical Computing in Python

Andrew Cron
Duke University

Wes McKinney
Duke University


In this work we discuss gpustats, a new Python library for assisting in \textquotedbl{}big data\textquotedbl{} statistical computing applications, particularly Monte Carlo-based inference algorithms. The library provides a general code generation / metaprogramming framework for easily implementing discrete and continuous probability density functions and random variable samplers. These functions can be utilized to achieve more than 100x speedup over their CPU equivalents. We demonstrate their use in an Bayesian MCMC application and discuss avenues for future work.


GPU, CUDA, OpenCL, Python, statistical inference, statistics, metaprogramming, sampling, Markov Chain Monte Carlo (MCMC), PyMC, big data



Bibtex entry

Full text PDF