Feat/disable log gradients

[lorenzomammana]

Summary

Remove gradients logging callback from default configs to avoid slowing down the training process

Type of Change

Please select the one relevant option below:

• Refactor

Checklist

Please confirm that the following tasks have been completed:

• I have tested my changes locally and they work as expected. (Please describe the tests you performed.)
• I have added unit tests for my changes, or updated existing tests if necessary.
• I have updated the documentation, if applicable.
• I have installed pre-commit and run locally for my code changes.