Skip to content

Pull requests: NVIDIA/TransformerEngine

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

build: fix PyTorch wheel CUDA index calculation community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3118 opened Jun 10, 2026 by fallintoplace Contributor Loading…
3 of 11 tasks
2
[PyTorch] Refactor grouped MLP into joint forward-backward fused op MoE refactor
#3117 opened Jun 10, 2026 by timmoon10 Member Loading…
8 of 13 tasks
7
TE EP integration to MoEBlock
#3116 opened Jun 10, 2026 by tdophung Collaborator • Draft
13 tasks
9
[JAX] Collective Gemm test fixes
#3115 opened Jun 10, 2026 by jberchtold-nvidia Collaborator Loading…
13 tasks
2
Current Scaling Group Quantization + Enabling Varying Last/Both Dims in Group Quantize
#3114 opened Jun 10, 2026 by vthumbe1503 Collaborator • Draft
13 tasks
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…
10
[JAX] Return max_logit and softmax aux stats from TE JAX fused attn 2.17
#3112 opened Jun 10, 2026 by KshitijLakhani Collaborator • Draft
13 tasks
1
@KshitijLakhani
[PyTorch] Refactor grouped linear and grouped MLP tests MoE refactor testing Improvements to tests or testing infrastructure
#3111 opened Jun 10, 2026 by timmoon10 Member Loading…
9 of 14 tasks
5
[WIP][Common,pytorch] Add FlexQuantizer for customized quantization via CuTeDSL
#3110 opened Jun 10, 2026 by kainzhong Collaborator • Draft
13 tasks
1
[PyTorch] Add op-level activation offload opt-out API community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3108 opened Jun 9, 2026 by lhb8125 Contributor Loading…
19
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…
7
[PyTorch][torch.compile] Remove process group from quantizers
#3104 opened Jun 8, 2026 by pggPL Collaborator Loading…
3 of 12 tasks
8
Quantization support for GroupedTensor: FP8 per-tensor community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3102 opened Jun 7, 2026 by int-smart Contributor Loading…
11 of 13 tasks
1
6
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 • Draft
1 of 15 tasks
9
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
NVFP4: cache GEMM-swizzled weight scale factors across micro-batches community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3093 opened Jun 5, 2026 by cael-ling Contributor Loading…
3 of 13 tasks
2
Added thd cudnn guard community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3092 opened Jun 5, 2026 by francesco-bertolotti Contributor Loading…
6 of 13 tasks
5
guarding max_logits fused attention for cudnn < 9.21.0 community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3091 opened Jun 5, 2026 by francesco-bertolotti Contributor Loading…
6 of 13 tasks
5
Make NVTE tensor handle pool size configurable community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3090 opened Jun 5, 2026 by lhb8125 Contributor • Draft
6
fix(topk): fix UB and prevent vector load splitting in standalone_topk community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3088 opened Jun 5, 2026 by solos Loading…
5 of 13 tasks
4
[JAX] Extend tensor inspect utility to dump out tensors in identifiable names
#3086 opened Jun 4, 2026 by tdophung Collaborator Loading…
6 of 13 tasks
5
[JAX] Fix norm workspace on global shapes
#3085 opened Jun 4, 2026 by jberchtold-nvidia Collaborator • Draft
8 of 13 tasks
6
[JAX] MoEBlock tutorial
#3084 opened Jun 4, 2026 by jberchtold-nvidia Collaborator • Draft
13 tasks
5
[JAX] Hopper BF16 grouped GEMM v2 support
#3083 opened Jun 4, 2026 by jberchtold-nvidia Collaborator • Draft
8 of 13 tasks
4
add attention docs
#3081 opened Jun 4, 2026 by sudhakarsingh27 Member • Draft
13 tasks
2
ProTip! What’s not been updated in a month: updated:<2026-05-10.