Link: https://github.com/unslothai/unsloth/blob/main/unsloth/kernels/cross_entropy_loss.py
Author: Daniel Han-Chen & the Unsloth team
Tags: Loss
Description:
- cross_entropy_forward
- chunked_cross_entropy_forward
- cross_entropy_backward
Triton Version: ?
Other Notes:
-
Id in triton index: 0003