Add parameter scaled by weight decay.
JIT-compatible with both scalar and schedule weightDecay.
weightDecay
Add parameter scaled by weight decay.
JIT-compatible with both scalar and schedule
weightDecay.