SGD implementation w/o L2 norm