Skip to content

Pull requests: ROCm/TransformerEngine

New pull request New
26 Open 451 Closed
26 Open 451 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

[WIP] NVFP4 Hadamard
#509 opened Mar 27, 2026 by matthiasdiener • Draft
13 tasks
@matthiasdiener
[TE] Enable deterministic mode for fused attention ci-level 1 CI test level 1
#508 opened Mar 27, 2026 by AllenFarcas Loading…
7 of 13 tasks
24
Mxfp8 cast optimization ci-level 3 CI test level 3
#507 opened Mar 26, 2026 by alextmagro Loading…
15
Full MXFP4 Training Recipe
#506 opened Mar 26, 2026 by sarthak-amd Loading…
2
NVFP4 dequantization
#505 opened Mar 25, 2026 by aris134 Loading…
6 of 13 tasks
@aris134
6
WIP IFU 2.12.
#496 opened Mar 18, 2026 by Micky774 Loading…
13 tasks
54
Add fsdp2 fp8 unit tests TE 2.10
#492 opened Mar 17, 2026 by sudhu2k Loading…
8 of 13 tasks
@sudhu2k
19
Add AITER fused RoPE dispatch to FusedRoPEFunc
#489 opened Mar 17, 2026 by sarthak-amd Loading…
14
ASV-format microbenchmark suite
#487 opened Mar 16, 2026 by Micky774 Loading…
1 of 13 tasks
10
Add Claude to review PRs
#480 opened Mar 13, 2026 by wenchenvincent Loading…
1 of 6 tasks
7
Microbenchmarking and CI performance regression test
#478 opened Mar 10, 2026 by matthiasdiener • Draft
4 of 17 tasks
@matthiasdiener
18
Utilize AITER HSA embeddings
#477 opened Mar 9, 2026 by Micky774 Loading…
1 of 13 tasks
15
Add FP8 Support For CK Tile Group GEMM
#475 opened Mar 6, 2026 by aris134 Loading…
7 of 15 tasks
@aris134
24
Use static libraries for AITER build
#468 opened Mar 2, 2026 by Micky774 Loading…
1 of 13 tasks
14
[NO MERGE] Integrate CK varlen cross attention for small-seq (s_q=1, s_kv<=16)
#461 opened Feb 24, 2026 by VeeraRajasekhar Loading…
13 tasks
@VeeraRajasekhar
127
Added initial AI Agent instructions and skills
#448 opened Feb 12, 2026 by Micky774 Loading…
2 of 13 tasks
5
Update README.rst
#435 opened Jan 28, 2026 by aris134 • Draft
1 of 13 tasks
Hotfix/fused ce triton
#409 opened Jan 12, 2026 by sarthak-amd Loading…
7
CI: Switch GHA pipeline to build and test wheels
#400 opened Dec 9, 2025 by leo-amd • Draft
1 of 6 tasks
2
Layernorm forward optimization
#377 opened Nov 24, 2025 by eliotwang Loading…
13 tasks
Userbuffer epic ci-level 3 CI test level 3
#367 opened Nov 11, 2025 by alextmagro Loading…
109
Fused Cross Entropy Triton - Loss Scaling and Vanishing Grads Bugfix
#336 opened Oct 16, 2025 by sarthak-amd Loading…
1 task done
8
heyi's layernorm optimization
#225 opened Jul 3, 2025 by eliotwang Loading…
8 of 13 tasks
@charyang-ai
[ROCm] support triton-based flash-attn in TE
#177 opened May 1, 2025 by wangye805 Loading…
8 of 13 tasks
6
Update attention example attention.ipynb
#152 opened Mar 19, 2025 by anhminhnguyenhoang Loading…
5 of 13 tasks
1
ProTip! Mix and match filters to narrow down what you’re looking for.