Skip to content

[JAX] Consolidate the distributed fused attention test code#1405

Open
mgoldfarb-nvidia wants to merge 1 commit intoNVIDIA:mainfrom mgoldfarb-nvidia:mgoldfarb-nvidia/consolidate_fused_attn_tests_code
+313-403

Commits

Commits on Jan 14, 2025