Function std::webgpu_sgd_update

#include <std/ml/gpu_webgpu.h> source · definition

int webgpu_sgd_update(struct WebGpuBuf w, struct WebGpuBuf g, float lr, unsigned long n)

w[i] -= lr * g[i] for i < n, in place (any dtypes). Queued.