@staticmethod
def _backward(loss):
    """
        Calls backward on loss tensor. Used to unify the API for PyTorch and apex optimizers
        :param loss: loss tensor
        :return: None
        """
    loss.backward()
