Skip to content

Pull requests: ROCm/TransformerEngine

New pull request New
40 Open 529 Closed
40 Open 529 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

TE AITER gfx1250 integration WIP
#603 opened May 29, 2026 by Micky774 Contributor • Draft
13 tasks
upgrade ci to TheRock ci-level 3 CI test level 3
#602 opened May 29, 2026 by VeeraRajasekhar Contributor • Draft
13 tasks
@VeeraRajasekhar
Update QoLA/AITER ci-level 3 CI test level 3
#599 opened May 28, 2026 by Micky774 Contributor Loading…
13 tasks
7
Mxfp8 grouped and multi quantize ci-level 3 CI test level 3
#598 opened May 27, 2026 by alextmagro Contributor Loading…
2
Fix CK FP8 grouped GEMM dtype gating for columnwise operands ci-level 1 CI test level 1
#594 opened May 21, 2026 by aris134 Contributor Loading…
1 of 12 tasks
@aris134
Triton RMSNorm Optimizations ci-level 3 CI test level 3
#593 opened May 20, 2026 by Micky774 Contributor Loading…
7 of 13 tasks
30
speed up nvte_multi_padding / nvte_multi_unpadding ci-level 1 CI test level 1
#592 opened May 20, 2026 by matthiasdiener Contributor Loading…
13 tasks
@matthiasdiener
Bump CI retention days ci-level 1 CI test level 1
#591 opened May 20, 2026 by matthiasdiener Contributor • Draft
1 of 13 tasks
@matthiasdiener
9
add production GEMM tests ci-level 1 CI test level 1
#590 opened May 19, 2026 by matthiasdiener Contributor Loading…
1 of 13 tasks
@matthiasdiener
7
MXFP8 training bug fixes for quantized_model_init and Torch FSDP fp8 all gather ci-level 3 CI test level 3
#587 opened May 15, 2026 by sudhu2k Contributor Loading…
8 of 13 tasks
@sudhu2k
6
Optimized rocm specific multicast transpose kernel ci-level 3 CI test level 3
#586 opened May 14, 2026 by alextmagro Contributor Loading…
9
Add custom multi_tensor_apply kernels (L2norm, Adam) ci-level 1 CI test level 1
#585 opened May 13, 2026 by matthiasdiener Contributor • Draft
1 of 13 tasks
@matthiasdiener
15
CK JIT integration ci-level 1 CI test level 1
#582 opened May 11, 2026 by ipanfilo Collaborator Loading…
1 of 13 tasks
12
Add Tealite: pure-Python TransformerEngine for ROCm/AMD GPUs
#581 opened May 7, 2026 by jayfurmanek Contributor Loading…
7 of 8 tasks
ck_tile grouped gemm: more padding ci-level 3 CI test level 3
#574 opened May 5, 2026 by matthiasdiener Contributor Loading…
1 of 13 tasks
@matthiasdiener
29
[ROCm] Allow bf16/bf16/fp32 in nvte_multi_tensor_gemm dispatcher ci-level 1 CI test level 1
#573 opened May 4, 2026 by lizamd Loading…
13 tasks
9
[No Merge][No Review] testing aiter auto trigger on gh action ci-level 2 CI test level 2
#570 opened May 1, 2026 by VeeraRajasekhar Contributor • Draft
13 tasks
@VeeraRajasekhar
HipKittens MXFP8 GEMM Support ci-level 3 CI test level 3
#566 opened Apr 28, 2026 by alextmagro Contributor Loading…
115
[WIP] TDM porting
#558 opened Apr 22, 2026 by wangye805 Collaborator • Draft
13 tasks
83
IFU v2.14.dev0 ci-level 3 CI test level 3
#557 opened Apr 21, 2026 by ipanfilo Collaborator Loading…
5 of 13 tasks
10
Enable CI lint gh action on ROCm ci-level 3 CI test level 3
#547 opened Apr 17, 2026 by VeeraRajasekhar Contributor Loading…
13 tasks
@VeeraRajasekhar
27
CI: auto-trigger AITER prebuilt upload when 3rdparty/aiter updates on dev
#543 opened Apr 15, 2026 by VeeraRajasekhar Contributor Loading…
8 of 13 tasks
@VeeraRajasekhar
18
[TE] Phase 2 of small-seq cross-attn integration: a separate cpp backend and a new jax api ci-level 3 CI test level 3
#542 opened Apr 15, 2026 by VeeraRajasekhar Contributor Loading…
13 tasks
@VeeraRajasekhar
35
Integrate AITER fused RoPE kernels with fallback to TE native
#541 opened Apr 15, 2026 by suachong Contributor Loading…
7 tasks done
32
NVFP4: hadamard_transform_cast_fusion_columnwise ci-level 1 CI test level 1
#515 opened Apr 1, 2026 by matthiasdiener Contributor • Draft
1 of 13 tasks
@matthiasdiener
1
ProTip! no:milestone will show everything without a milestone.