Hello 
I’m trying to implement a function which operates on the raw model loss at backward time due to badly defined gradients. One naive approach is to define a global variable which contains the loss and access it as is, however playing around with global variables is a bit edgy.
As such, is there a mecanism that allows to get the model loss from the ctx in a few lines ? (i.e. not modifying the whole model)