megengine.optimizer.AdamW.backward

AdamW.backward(loss)