Code to run a regression on the GPU