Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
135 Open 2,271 Closed
135 Open 2,271 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

[JAX] Keep the routing map format alive and EP multiprocess tests in L2 Jax dist 2.17 urgent
#3159 opened Jul 1, 2026 by KshitijLakhani Collaborator Loading…
4 of 13 tasks
@KshitijLakhani
3
Migrate norms and softmax kernels to NVRTC community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3156 opened Jun 30, 2026 by CarlosGomes98 Contributor Loading…
2 of 13 tasks
7
[PyTorch][torch.compile] Add TensorProto mechanism
#3153 opened Jun 29, 2026 by pggPL Collaborator Loading…
4 of 13 tasks
4
[PyTorch][torch.compile] Make quantizers opaque value objects
#3152 opened Jun 29, 2026 by pggPL Collaborator Loading…
8 of 13 tasks
25
Enable FA4 for context-parallel attention
#3149 opened Jun 26, 2026 by sudhakarsingh27 Member • Draft
7 of 13 tasks
[Draft] Use vendored cuDNN frontend for Python
#3148 opened Jun 26, 2026 by vcherepanov-nv Collaborator Loading…
1 of 13 tasks
6
Add MXFP8 support with cuBLASMp community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3145 opened Jun 25, 2026 by almogsegal Contributor Loading…
13 tasks
@denera
8
Add multi_tensor_raw_moments kernel community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3144 opened Jun 25, 2026 by philipcmonk • Draft
6 of 13 tasks
docs: document attention backend selection documentation Improvements or additions to documentation
#3142 opened Jun 24, 2026 by sbhavani Collaborator Loading…
4 of 13 tasks
1
[Common] Fix Build: NCCL EP build to respect MAX_JOBS
#3138 opened Jun 22, 2026 by phu0ngng Collaborator • Draft
7 of 13 tasks
3
[Common] Experimental CuTeDSL MXFP8 backends in C++ via TVM-FFI
#3137 opened Jun 21, 2026 by kainzhong Collaborator • Draft
13 tasks
[Common/PyTorch] Grouped-quantize kernels for 1D and 2D FP8 block-scaling 2.17 FP8 MoE performance Performance issues
#3135 opened Jun 17, 2026 by denera Collaborator Loading…
8 of 13 tasks
1
@denera
29
Single-launch CUTLASS grouped GEMM for per-tensor NVFP4 community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3134 opened Jun 17, 2026 by cael-ling Contributor Loading…
9 of 13 tasks
3
Enable NVFP4 RHT amax for grouped SReLU MLP community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3133 opened Jun 16, 2026 by sraman-rgb Contributor Loading…
13 tasks
11
[Common] Support scaled & clamped swiglu, srelu for BF16 community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3132 opened Jun 16, 2026 by zhongbozhu Collaborator Loading…
13 tasks
9
[JAX] Remove shard_map from MoEBlock to support quant before FSDP AG using Grouped quant+GEMM custom partitioning rules
#3131 opened Jun 15, 2026 by jberchtold-nvidia Collaborator • Draft
13 tasks
5
feat: add SM_121 (GB10 consumer Blackwell) support for FA4 community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3125 opened Jun 12, 2026 by TyGu1 Loading…
5
docs(readme): update latest news documentation Improvements or additions to documentation
#3121 opened Jun 11, 2026 by sbhavani Collaborator Loading…
6 of 13 tasks
1
TE EP integration to MoEBlock
#3116 opened Jun 10, 2026 by tdophung Collaborator Loading…
6 of 13 tasks
27
[JAX] Collective Gemm test fixes
#3115 opened Jun 10, 2026 by jberchtold-nvidia Collaborator Loading…
13 tasks
2
Abstract CUDA hardcodes into configurable te_device_type / te_platform community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3113 opened Jun 10, 2026 by lxd-cumt Loading…
19
[JAX] Return max_logit and softmax aux stats from TE JAX fused attn 2.18
#3112 opened Jun 10, 2026 by KshitijLakhani Collaborator • Draft
13 tasks
1
@KshitijLakhani
Add entrypoint for flagos multi-backend plugin system community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3107 opened Jun 9, 2026 by lxd-cumt Loading…
15
Introduce Mega-C++ to reduce CPU overhead community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3099 opened Jun 6, 2026 by zhongbozhu Collaborator Loading…
3 of 17 tasks
17
increased a bit tolerance for pytorch/distributed/run_numerics.py community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3095 opened Jun 5, 2026 by francesco-bertolotti Contributor Loading…
6 of 13 tasks
3
ProTip! no:milestone will show everything without a milestone.