Skip to content

Pull requests: ROCm/TransformerEngine

New pull request New
22 Open 372 Closed
22 Open 372 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

Resolved conflicts
#425 opened Jan 22, 2026 by VeeraRajasekhar • Draft
13 tasks
@VeeraRajasekhar
Enable MXFP8 support in TE JAX integration
#424 opened Jan 20, 2026 by Micky774 Loading…
13 tasks
Release v2.8 rocm
#423 opened Jan 20, 2026 by VeeraRajasekhar • Draft
13 tasks
@VeeraRajasekhar
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
26
Enhance GroupedLinear with integrating AITER triton kernels
#413 opened Jan 15, 2026 by sudhu2k Loading…
8 of 13 tasks
@sudhu2k
25
[CI] Automate AITER prebuilt upload flow with GitHub actions
#412 opened Jan 14, 2026 by VeeraRajasekhar Loading…
8 of 13 tasks
@VeeraRajasekhar
31
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
Enable JAX distributed test for THD+Ring
#405 opened Jan 2, 2026 by Micky774 Loading…
13 tasks
2
CI: Switch GHA pipeline to build and test wheels
#400 opened Dec 9, 2025 by leo-amd • Draft
1 of 6 tasks
2
GEMM reference computation offload
#392 opened Dec 4, 2025 by matthiasdiener Loading…
1 of 13 tasks
@matthiasdiener
18
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 • Draft
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
14
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! Mix and match filters to narrow down what you’re looking for.