WebGPU Linear Regression Training