Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
198 Open 2,553 Closed
198 Open 2,553 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

KDA attention 2.20
#3548 opened Sep 18, 2026 by ksivaman Member Loading…
8 of 13 tasks
3
[PyTorch] Compile Linear and Bias operations with forward fusion
#3547 opened Sep 18, 2026 by pggPL Collaborator • Draft
6 of 9 tasks
[PyTorch] Enable LayerNormLinear UB AG overlap under no_grad community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3546 opened Sep 18, 2026 by ravimajeti Loading…
7 of 8 tasks
1
1
[PyTorch] Keep DistributedWeight objects on ctx across saved-tensor hooks community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3545 opened Sep 18, 2026 by xrennvidia Collaborator Loading…
1 of 13 tasks
2
feat: vmm slot for local cuda graph offload community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3544 opened Sep 18, 2026 by Xdydy Loading…
1 of 13 tasks
6
Use PyTorch CUTLASS for BF16 grouped GEMM on SM100 community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3542 opened Sep 18, 2026 by wujingyue Contributor • Draft
5 of 7 tasks
5
Fix distributed Newton-Schulz correctness 2.20
#3541 opened Sep 17, 2026 by vcherepanov-nv Collaborator Loading…
4 of 13 tasks
4
[PyTorch][CP] Support compact metadata with FA4
#3540 opened Sep 17, 2026 by sudhakarsingh27 Member Loading…
13 tasks
3
fix: add missing NVTE_BHSD labels to fused_attn to_string community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3538 opened Sep 17, 2026 by andrewwhitecdw Contributor Loading…
1
Prototype of green context + VMM localization community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3537 opened Sep 17, 2026 by WanZzzzzz Contributor • Draft
13 tasks
test(attention): anchor the context-parallel suite to an independent reference community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3536 opened Sep 17, 2026 by nvegesna-netizen Contributor Loading…
2
[Docs] Refresh README
#3535 opened Sep 17, 2026 by sbhavani Collaborator Loading…
5 of 13 tasks
1
[PyTorch] [torch.compile] torch.compile support for LayerNormLinear and LayerNormMLP
#3534 opened Sep 17, 2026 by pggPL Collaborator Loading…
2
[PyTorch] Retain NVFP4 RNG tensors through quantization community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3533 opened Sep 17, 2026 by Connor-XY Loading…
1
fix(attention): stop handing FlashAttention 4 the -1 window sentinel community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3532 opened Sep 17, 2026 by nvegesna-netizen Contributor Loading…
1
4
[PyTorch] Extend no-load-balance CP to the a2a comm type community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3530 opened Sep 16, 2026 by Rudin6 Loading…
13 tasks
@sudhakarsingh27
2
feat(attention): cuDNN FROST attention backend for head_dim in (256, 512], with context parallelism 2.21.0 community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3527 opened Sep 16, 2026 by nvegesna-netizen Contributor Loading…
17
[Pytorch][NCCL EP] Create tokens_per_expert on pinned CPU memory when running eager
#3526 opened Sep 16, 2026 by YangFei1990 Collaborator Loading…
8 of 13 tasks
4
[JAX] WAR for XLA associative scan rewriter crash 2.20
#3525 opened Sep 16, 2026 by KshitijLakhani Collaborator Loading…
2 of 13 tasks
@KshitijLakhani
5
Remove the view() calls from Linear/LNLinear/LNMLP modules
#3523 opened Sep 15, 2026 by ptrendx Member Loading…
1 of 13 tasks
6
[JAX] Support single-process-multi-device EP via XLA-borrowed communicator
#3522 opened Sep 15, 2026 by phu0ngng Collaborator Loading…
8 of 13 tasks
6
Accept compact MXFP8 scales in GEMM preparation community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3519 opened Sep 15, 2026 by wujingyue Contributor • Draft
1
4
[PyTorch] Support distributed weights in GroupedLinear's grouped-tensor path org-contribution
#3517 opened Sep 15, 2026 by fanshiqing Member Loading…
7 of 8 tasks
1
11
[JAX] Align fused attn backward output gradient sharding and other test strengthening 2.20 attention bug Something isn't working
#3516 opened Sep 14, 2026 by KshitijLakhani Collaborator Loading…
1 of 13 tasks
@KshitijLakhani
7
Support building docs for publishing via GHA
#3515 opened Sep 14, 2026 by fheinecke Collaborator Loading…
1 of 13 tasks
2
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.