How can I customize the gradient computation at training time in keras?

Question

I want to implement the natural gradient in a keras layer. This has to happen inside the customized gradient that is already in place. I want to be able to choose which impl