Log gradient norms to understand and debug model better #688
Description
Is your feature request related to a problem? Please describe.
See title, ideally with some plotting code, and potentially behind a flag to prevent large metric files.
Describe the solution you'd like
No response
Describe alternatives you've considered
No response
Additional context
No response
Organisation
No response