optimizer.update(model, grads)