Hugging Face's logo

flashrt
/
flashrt-flex-attention-train

Model card Files
xet
 Community
flashrt-flex-attention-train / torch-ext /flashrt_flex_attention_train

Commit History

  1. Add manual_attention_part_hp (fp32-logits variant)

    liangsu9988
    Verified signature
    af9e65b

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

    liangsu9988
    Verified signature
    285871e

  2. Add package source and torch-universal build variant

    liangsu9988
    Verified signature
    b22e03e