Skip to content

Pull requests: ROCm/TransformerEngine

New pull request New
23 Open 383 Closed
23 Open 383 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

RMSNorm alignment calculation hotfix
#436 opened Jan 28, 2026 by Micky774 Loading…
13 tasks
Update README.rst
#435 opened Jan 28, 2026 by aris134 • Draft
1 of 13 tasks
[WIP] Grouped GEMM with ck_tile
#434 opened Jan 28, 2026 by matthiasdiener • Draft
2 of 16 tasks
@matthiasdiener
Add a README on how to run with act documentation Improvements or additions to documentation
#432 opened Jan 26, 2026 by matthiasdiener Loading…
2 of 13 tasks
@matthiasdiener
2
[DO NOT MERGE] [CI TEST] Run on any runner
#430 opened Jan 26, 2026 by leo-amd Loading…
release v2.8 rocm
#428 opened Jan 22, 2026 by VeeraRajasekhar • Draft
13 tasks
Enable MXFP8 support in TE JAX integration
#424 opened Jan 20, 2026 by Micky774 Loading…
13 tasks
52
MXFP4 Cast Transpose Triton [WIP]
#422 opened Jan 20, 2026 by sarthak-amd • Draft
1 task done
GEMMTestSuite: perform input data generation on GPU (incl. hiprand)
#417 opened Jan 16, 2026 by matthiasdiener Loading…
1 of 13 tasks
@matthiasdiener
34
Enhance GroupedLinear with integrating AITER triton kernels
#413 opened Jan 15, 2026 by sudhu2k Loading…
8 of 13 tasks
@sudhu2k
40
Hotfix/fused ce triton
#409 opened Jan 12, 2026 by sarthak-amd Loading…
6
IFU release v2.6
#406 opened Jan 3, 2026 by wangye805 Loading…
10 of 13 tasks
6
CI: Switch GHA pipeline to build and test wheels
#400 opened Dec 9, 2025 by leo-amd • Draft
1 of 6 tasks
2
Enable AOTriton BWD V3 API
#382 opened Nov 25, 2025 by Micky774 Loading…
13 tasks
@Micky774
7
Layernorm forward optimization
#377 opened Nov 24, 2025 by eliotwang Loading…
13 tasks
Userbuffer epic
#367 opened Nov 11, 2025 by alextmagro Loading…
Fused Cross Entropy Triton - Loss Scaling and Vanishing Grads Bugfix
#336 opened Oct 16, 2025 by sarthak-amd Loading…
1 task done
8
Triton norms dispatch refactor
#305 opened Sep 5, 2025 by Micky774 Loading…
13 tasks
15
heyi's layernorm optimization
#225 opened Jul 3, 2025 by eliotwang Loading…
8 of 13 tasks
@charyang-ai
Added Dockerfile for CI images
#195 opened May 28, 2025 by VeeraRajasekhar Loading…
7 of 13 tasks
31
[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
Honor the NVTE_FUSED_ATTN_<backend> in test_fused_attn.py
#123 opened Feb 11, 2025 by wangye805 Loading…
13 tasks
3
ProTip! no:milestone will show everything without a milestone.