Generalize SGD to use .params instead of .W and .b