transitioning to use gpu for backprop :-)