7 people 1dee77f4a4 [feat] QAD 5090: Wire the Attn-QAT training attention backend (10/12) (#1459) ...

Co-authored-by: William Lin <SolitaryThinker@users.noreply.github.com>
Co-authored-by: Loay Rashid <42599591+loaydatrain@users.noreply.github.com>
Co-authored-by: Kaiqin Kong <k1kong@ucsd.edu>
Co-authored-by: SolitaryThinker <wlsaidhi@gmail.com>
Co-authored-by: alexzms <26690162+alexzms@users.noreply.github.com>
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>

2026-06-17 13:32:56 -07:00

..

__init__.py

[Feature] Remove V1 folder (#642)

2025-07-24 22:43:12 -07:00

abstract.py

[refactor]: shared attention infra additions for QAT-compat (Attn-QAT 5/12) (#1383)

2026-05-23 16:19:24 -07:00

attn_qat_infer.py

[feat]: Attn-QAT inference + training backends (deadcode) (Attn-QAT 4/12) (#1358)

2026-05-22 02:59:25 -07:00

attn_qat_train.py

[feat] QAD 5090: Wire the Attn-QAT training attention backend (10/12) (#1459)

2026-06-17 13:32:56 -07:00

bsa_attn.py

[refactor]: shared attention infra additions for QAT-compat (Attn-QAT 5/12) (#1383)

2026-05-23 16:19:24 -07:00

flash_attn.py

[perf]: register FA2/FA3 default flash_attn_func as a torch.library custom op (#1373)

2026-05-23 00:39:06 -07:00

sage_attn3.py

[feat] pre-commit support 120 col num (#1167)

2026-03-11 20:19:30 -07:00

sage_attn.py

[Feature] Remove V1 folder (#642)

2025-07-24 22:43:12 -07:00

sdpa.py

[feat] pre-commit support 120 col num (#1167)

2026-03-11 20:19:30 -07:00

sla.py

[misc] attention hot-path cleanup + denoising loop hoists (#1272)

2026-05-10 11:51:10 -07:00

video_sparse_attn.py

[bugfix]: release VSA tile cache during training (#1434)

2026-06-09 15:01:22 -07:00

vmoba.py

[refactor]: shared attention infra additions for QAT-compat (Attn-QAT 5/12) (#1383)

2026-05-23 16:19:24 -07:00