Hugging Face's logo

Kernels:
trl-lib
/
fused-linear-ce

Kernel card Files
xet
 Community
fused-linear-ce / build

Commit History

  1. Metrics return sums (num_correct_tokens, entropy_sum) instead of means

    qgallouedec HF Staff
    Verified signature
    00e29bd

  2. Clean-room Triton fused linear cross-entropy kernel

    qgallouedec HF Staff
    Verified signature
    cc8550e