[Feature] Add on_after_backward in Callback. Enable ModelGradTrackerCallback
🚀 Feature

Motivation

The call_hook on_after_backward  is already implemented, but not added in Callback class.
It boils down to add it within Callback Class
Pitch

Adding this new hook within callback could be used to implement something likeModelGradTrackerCallback.
Alternatives

Additional context