on_after_backward for multiple optimizer
🚀 Feature
on_after_backward is a perfect place to log grad, currently, on_after_backward make no distinguish for different optimizers, for GAN applications, in it would be nice to pass in optimizer_id as param for on_after_backward.