First working version : >4000x faster than CPU, no GPU optimization, MSE of ~0.98