flashrt
/

flashrt-flex-attention-train

Model card Files Files and versions

flashrt-flex-attention-train / tests

8.84 kB

Ctrl+K

Ctrl+K

1 contributor

History: 2 commits

liangsu9988's picture

Add manual_attention_part_v2 (bf16-saved-p, fused softmax gradient)

285871e verified 18 days ago

test_flashrt_flex_attention_train.py

8.84 kB
Add manual_attention_part_v2 (bf16-saved-p, fused softmax gradient) 18 days ago