Currently, `coix.loss` returns surrogate objectives. It would be nice to return original objective rather than surrogate loss. ``` return surrogate_loss + stop_gradident(original_loss - surrogate_loss) ```