Got SGD to work, very slow though, maybe GPU version would be better?