Skip to content

[ROCm] Jax Add softmax sink (learnable off-by-one) support for the ROCm/CK fused attention backend - #678

Open
shurale-nkn wants to merge 14 commits into
devfrom
sink_support_ck
Open

[ROCm] Jax Add softmax sink (learnable off-by-one) support for the ROCm/CK fused attention backend#678
shurale-nkn wants to merge 14 commits into
devfrom
sink_support_ck

Commits

Commits on Jul 9, 2026

Commits on Jul 15, 2026

Commits on Jul 24, 2026

Commits on Jul 26, 2026

Commits on Jul 27, 2026

Commits on Jul 29, 2026

Commits on Jul 30, 2026

Commits on Jul 31, 2026

Commits on Aug 3, 2026