Add a gradient logger to WTFML

[RJaikanth]

I have created a function to log the gradients if they are vanishing or exploding. Would like to contribute the function here.

@abhishekkrthakur Let me know if I can create a PR for the same that is if you want to add this feature.