Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
155 Open 2,311 Closed
155 Open 2,311 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

Add stream-ordered CP gradient return primitive community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3235 opened Jul 22, 2026 by foraxe Loading…
5 of 13 tasks
[PyTorch] Optionally release columnwise copy of frozen FP8 block-scaled weights after dgrad community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3233 opened Jul 22, 2026 by 1tex • Draft
8 of 13 tasks
Enable Weight Preswizzling only when Swizzle fusion is available
#3232 opened Jul 21, 2026 by vthumbe1503 Collaborator Loading…
13 tasks
1
12
[JAX] Schedule EP dispatch/combine on XLA collective stream
#3231 opened Jul 21, 2026 by phu0ngng Collaborator Loading…
9 of 13 tasks
6
[CI] Fix old NVCC dependency in GH CI
#3230 opened Jul 21, 2026 by fheinecke Collaborator • Draft
5 of 13 tasks
[PyTorch] NCCL EP eager mode and drop-on-overflow policy
#3229 opened Jul 21, 2026 by phu0ngng Collaborator • Draft
8 of 13 tasks
[CI] Minor CMake refactoring
#3227 opened Jul 21, 2026 by fheinecke Collaborator • Draft
5 of 13 tasks
1
[JAX] Support axes orthogonal to EP in ep_bootstrap via mesh-derived domain grouping
#3226 opened Jul 21, 2026 by phu0ngng Collaborator Loading…
8 of 13 tasks
4
Single grouped weight fixes community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3225 opened Jul 21, 2026 by CarlosGomes98 Contributor Loading…
13 tasks
3
Improve device-init grouped linear module with single grouped weight support community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3224 opened Jul 20, 2026 by zhongbozhu Collaborator • Draft
13 tasks
2
[Common] Experimental CuTeDSL MXFP4 backend community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3223 opened Jul 20, 2026 by janekb04 Collaborator • Draft
1 of 13 tasks
[Common] Migrate NCCL EP submodule to NVIDIA/nccl-extensions
#3222 opened Jul 20, 2026 by phu0ngng Collaborator Loading…
6 of 13 tasks
4
[PyTorch] Add experimental packed-contiguous THD all-gather CP
#3221 opened Jul 18, 2026 by sudhakarsingh27 Member • Draft
Fix SM120 FP8 output ordering before DelayedScaling amax update community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3215 opened Jul 15, 2026 by AlbertYang514 Loading…
5
Explore error correction for pertoken recipe community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3214 opened Jul 15, 2026 by YigongQin Contributor • Draft
13 tasks
Fix duplicate FP8 state updates during activation checkpoint recomputation community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3213 opened Jul 15, 2026 by AlbertYang514 Loading…
7 of 13 tasks
5
[PyTorch][common] NVFP4: enable row-scaled transpose quantization for backward community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3206 opened Jul 13, 2026 by cael-ling Contributor Loading…
13 tasks
10
[PyTorch] Support padded THD sequences with FlashAttention 4 community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3205 opened Jul 13, 2026 by xiaoyao0115 • Draft
7 tasks done
[PyTorch][torch.compile] Support for UnfusedDotProductAttention
#3201 opened Jul 10, 2026 by pggPL Collaborator Loading…
7 of 13 tasks
2
[PyTorch][torch.compile] DotProductAttention: declarative packed QKV/KV inputs
#3200 opened Jul 10, 2026 by pggPL Collaborator Loading…
8 of 13 tasks
7
Developer documentation
#3198 opened Jul 9, 2026 by ptrendx Member • Draft
1 of 13 tasks
Add transformer_engine.test() installation smoke check community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3197 opened Jul 9, 2026 by CodersAcademy006 Loading…
1
5
docs: add datatype and hardware support matrix community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3195 opened Jul 9, 2026 by CodersAcademy006 Loading…
2
1
adam: separate param/grad dtypes in AdamCapturableFunctor community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3192 opened Jul 8, 2026 by rapatel Loading…
5 of 13 tasks
2
[PyTorch][torch.compile] Make get_attention_backend traceable without graph breaks
#3189 opened Jul 8, 2026 by pggPL Collaborator Loading…
8 of 13 tasks
5
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.