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