3 people 9709b7513b [feat] Port NVFP4 QAT/QAD to modular train framework (#1619) ...

Co-authored-by: Peiyuan Zhang <email>

Co-authored-by: Peiyuan <a>

2026-07-18 15:01:09 -07:00

..

__init__.py

[kernel] Reorg and fix fastvideo-kernel (#962)

2025-12-26 01:50:24 -06:00

support_flex_sta.py

[kernel] Reorg and fix fastvideo-kernel (#962)

2025-12-26 01:50:24 -06:00

test_attn_qat_infer.py

[infra] Add DGX Spark and multi-architecture CUDA support (#1447)

2026-07-03 13:41:43 -07:00

test_attn_qat_train_routing.py

[feat] Port NVFP4 QAT/QAD to modular train framework (#1619)

2026-07-18 15:01:09 -07:00

test_attn_qat_train.py

[kernel] QAD 5090: Add Attn-QAT training Triton kernels (11/12) (#1460)

2026-06-17 13:35:12 -07:00

test_fused_compress_topk.py

[perf] optimize compress & topk kernel (#1421)

2026-06-23 18:51:22 +00:00

test_sta.py

[chore] release v0.1.7 (#955)

2025-12-27 05:05:37 -06:00

test_turbodiffusion.py

[kernel] add turbodiffusion kernels (#972)

2025-12-30 04:23:10 -06:00

test_vmoba_correctness.py

[kernel] Reorg and fix fastvideo-kernel (#962)

2025-12-26 01:50:24 -06:00

test_vsa256_forward_cross.py

[feat] VSA-256 fastpath on Blackwell via FA4 CuTe block-sparse attention (#1354)

2026-05-26 12:58:29 -07:00

test_vsa256_forward_vbs.py

[feat] VSA-256 fastpath on Blackwell via FA4 CuTe block-sparse attention (#1354)

2026-05-26 12:58:29 -07:00

test_vsa256_forward.py

[feat] VSA-256 fastpath on Blackwell via FA4 CuTe block-sparse attention (#1354)

2026-05-26 12:58:29 -07:00

test_vsa256_triton.py

[feat] VSA-256 fastpath on Blackwell via FA4 CuTe block-sparse attention (#1354)

2026-05-26 12:58:29 -07:00

test_vsa_forward.py

[Bug Fix] Add autograd wrapper for block-sparse attention in fastvideo-kernel + fix CMake extension linking (#1015)

2026-01-16 21:16:43 -08:00

test_vsa_utils.py

[kernel] Extract VSA utility functions into fastvideo_kernel (#1408)

2026-06-30 12:19:25 -07:00

test_vsa_varlen.py

[kernel] Add varlen support for block-sparse attention (#1319)

2026-06-18 01:06:00 +00:00

test_vsa.py

[Bug Fix] Add autograd wrapper for block-sparse attention in fastvideo-kernel + fix CMake extension linking (#1015)

2026-01-16 21:16:43 -08:00

utils.py

[chore] release v0.1.7 (#955)

2025-12-27 05:05:37 -06:00