Wing Lian 163bd4dd5a use custom triton kernels for entropy from logits and selective softmax (#3510) ...

* use custom triton kernels for entropy from logits and selective softmax

* PR comments fixes

* fix out of bounds, include tests, include benchmarks

* chore: lint

2026-03-19 02:02:43 -04:00

6.2 KiB

Raw Blame History

View Raw