How to add norm of gradient to a loss function?

Specific Domains Machine Learning
,
13

As always, please include a full stacktrace along with your MWE.