Implement L-inf regularization in SGD