def METHOD_NAME(self, loss: Tensor):
    """
    Scales the loss for mixed precision training.
    Note: Only available for optimizers with mixed precision training.
    Args:
        loss (Tensor): The loss to be scaled.
    """
    raise NotImplementedError(
        "The method scale_loss is only available for optimizers with mixed precision training")