Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
163 Open 2,374 Closed
163 Open 2,374 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

[Common/PyTorch] Grouped weighted-SwiGLU MXFP8 kernel community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3315 opened Aug 4, 2026 by cael-ling Contributor Loading…
3 of 13 tasks
3
Log when thd with dropout falls to the composite cuDNN engine community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3313 opened Aug 4, 2026 by bzantium Loading…
3
[CI] Publish GB200 aarch64 wheel artifacts community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3311 opened Aug 4, 2026 by bvolpato Loading…
5 of 13 tasks
3
CI only - do not merge: Add debugging info to figure out what broke nightly build
#3307 opened Aug 3, 2026 by fheinecke Collaborator • Draft
1
[JAX] Migrate EP collective-stream annotation to compute_on2
#3304 opened Aug 3, 2026 by phu0ngng Collaborator Loading…
8 of 13 tasks
3
Add support for fused Q Up-Proj GEMM/RoPE/Quant. community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3303 opened Aug 3, 2026 by chaseblock Contributor Loading…
1 of 8 tasks
10
nvrtc MXFP8 kernels community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3302 opened Aug 3, 2026 by CarlosGomes98 Contributor • Draft
13 tasks
NVRTC NVFP4 quantization kernels community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3301 opened Aug 3, 2026 by CarlosGomes98 Contributor Loading…
8 of 13 tasks
17
Add NVFP4 RHT Support for SM120 and SM121 community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3300 opened Aug 3, 2026 by new-TonyWang Loading…
12
[PyTorch] Scope the quantized-param caching flag to its own graph capture community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3298 opened Aug 1, 2026 by xiuhu17 Contributor Loading…
6 tasks done
1
[Pytorch][Common] Option to Disable 2nd level Scale in NVFP4
#3297 opened Jul 31, 2026 by vthumbe1503 Collaborator Loading…
13 tasks
3
[PTQ] Store FP32 global scaling factors (absmax or scale_inv) for all quantized activations and weights. org-contribution
#3296 opened Jul 31, 2026 by cspades Member Loading…
7 of 13 tasks
5
[Common] Fix pointer arithmatic to generate correct LDS/STS instructions
#3291 opened Jul 30, 2026 by kainzhong Collaborator Loading…
8 of 13 tasks
2
Enable THD with hierarchical context parallelism
#3290 opened Jul 30, 2026 by sudhakarsingh27 Member • Draft
2 of 13 tasks
@sudhakarsingh27
Fix cuDNN SDPA score_mod cache collision, sm_12x arch gates, per-device plan cache community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3289 opened Jul 30, 2026 by YangXu1990uiuc Loading…
1
[PyTorch][torch.compile] Support for DotProductAttention on flash and unfused backends
#3286 opened Jul 30, 2026 by pggPL Collaborator Loading…
9 tasks done
3
[PyTorch] Preserve FP8 recompute state for inner autocast community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3284 opened Jul 30, 2026 by nvegesna-netizen Loading…
@pggPL
9
[JAX] Allow input tensors to be sharded with TP in EP dispatch/combine ops
#3283 opened Jul 30, 2026 by phu0ngng Collaborator • Draft
13 tasks
[Common][PyTorch] EP Fused prepare and dispatch for non-eager mode
#3282 opened Jul 30, 2026 by phu0ngng Collaborator • Draft
13 tasks
[Common] Fix NVFP4 stochastic rounding on architectures without cvt.rs community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3281 opened Jul 29, 2026 by davidkny22 Loading…
6 of 13 tasks
@ptrendx
3
[Pytorch] [NCCL EP] Allow zero tokens for an EP rank in eager mode
#3276 opened Jul 29, 2026 by YangFei1990 Collaborator Loading…
7 of 13 tasks
4
Add per-sequence causal policy to packed THD attention community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3274 opened Jul 29, 2026 by desh2608 Loading…
7 of 13 tasks
2
Minimize the memory usage of the fused cross entropy kernel
#3273 opened Jul 29, 2026 by ptrendx Member • Draft
1 of 13 tasks
[Common] Ensure quantization kernels handle noop properly
#3271 opened Jul 28, 2026 by kainzhong Collaborator Loading…
1 of 13 tasks
2
[Common][PyTorch] EP dispatch with unfused MXFP8 quantization
#3270 opened Jul 28, 2026 by phu0ngng Collaborator Loading…
8 of 13 tasks
21
ProTip! Follow long discussions with comments:>50.