Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
141 Open 2,184 Closed
141 Open 2,184 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

[PyTorch] Use data shape for MXFP8 pointer swizzle org-contribution
#3070 opened Jun 1, 2026 by jepio Member Loading…
7 of 13 tasks
2
[Pytorch] Add variable-K Cutlass GroupGEMM for fine-grained MoE wgrad community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3069 opened Jun 1, 2026 by cassiewilliam Contributor Loading…
6 of 8 tasks
4
Add FP16 error modes for NVFP4 4over6 community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3068 opened Jun 1, 2026 by zianglih Contributor Loading…
9 of 13 tasks
3
[Fix] Fix CUTLASS grouped GEMM segfault for empty groups community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3067 opened Jun 1, 2026 by Baibaifan Loading…
3
[PyTorch] Propagate skip_fp8_weight_update in GroupedLinear during FP8 CUDA graph capture community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3065 opened May 31, 2026 by LeSingh1 Contributor Loading…
1
Add public NVFP4 quantize API community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3064 opened May 31, 2026 by zianglih Contributor Loading…
8 of 13 tasks
9
fix unfused padding causal sdpa community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3063 opened May 31, 2026 by hungryGeek16 Loading…
4
increasing precision tolerance community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3060 opened May 29, 2026 by francesco-bertolotti Contributor Loading…
4 of 13 tasks
4
[JAX] Grouped quant+GEMM custom partitioning rules
#3058 opened May 28, 2026 by jberchtold-nvidia Collaborator • Draft
13 tasks
3
[Common/PyTorch] bugfix: Token-linear fused RoPE impl. for THD tensors. community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3057 opened May 28, 2026 by plugyawn Loading…
7 of 13 tasks
1
@sudhakarsingh27
6
[JAX] [PyT] [Common] Enable D=256 BWD cuDNN fused attn for Blackwell CC 10.x
#3056 opened May 28, 2026 by KshitijLakhani Collaborator Loading…
7 of 13 tasks
@KshitijLakhani
7
[PyTorch] [torch.compile] torch.compile support for Linear
#3053 opened May 28, 2026 by pggPL Collaborator • Draft
13 tasks
[PyTorch] Propagate FP8 graph weight update flag in GroupedLinear community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3052 opened May 28, 2026 by allenphilipj Loading…
1
3
[PyTorch] Integrate the cuBLAS single GEMM MXFP8 NN, NT support for sm120
#3050 opened May 28, 2026 by KshitijLakhani Collaborator • Draft
7 of 13 tasks
1
Enable NVFP4 fused grouped MLP community-contribution PRs from external contributor outside the core maintainers, representing community-driven work. org-contribution
#3048 opened May 27, 2026 by sraman-rgb Contributor Loading…
1 of 13 tasks
38
Feat/selective offload on srelu fuser community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3047 opened May 27, 2026 by lhb8125 Contributor Loading…
13 tasks
5
Add NVFP4 per-token quantization recipe community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3045 opened May 26, 2026 by cael-ling Contributor • Draft
13 tasks
3
[PyTorch Debug] Add scale_inv_std stat and skip NVFP4 layers in LogFp8TensorStats
#3044 opened May 26, 2026 by pggPL Collaborator Loading…
3
docs: expand comm gemm overlap guidance community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3043 opened May 26, 2026 by omribz156 Loading…
5 of 13 tasks
1
4
Use cuDNN for row-scaled NVFP4 grouped GEMM community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3042 opened May 26, 2026 by zianglih Contributor • Draft
[PyTorch debug] FakeQuant: support Float8BlockScaling and fix MoE / w… community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3040 opened May 25, 2026 by shangxiaokang • Draft
13 tasks
2
[PyTorch] Python DType enum
#3039 opened May 22, 2026 by vthumbe1503 Collaborator Loading…
13 tasks
9
[JAX] Expert Parallelism: JAX primitives + VJPs
#3036 opened May 22, 2026 by phu0ngng Collaborator Loading…
8 of 13 tasks
30
[PyTorch] Expert Parallelism: PyTorch wrapper + autograd ops with symm-mem zero-copy
#3035 opened May 22, 2026 by phu0ngng Collaborator • Draft
8 of 13 tasks
16
Expert Parallelism: common C API + NCCL EP backend
#3034 opened May 22, 2026 by phu0ngng Collaborator Loading…
8 of 13 tasks
37
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.