Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
179 Open 2,415 Closed
179 Open 2,415 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] Move ffi type definition before ffi target definitions 2.19
#3382 opened Aug 14, 2026 by jberchtold-nvidia Collaborator Loading…
8 of 13 tasks
2
[PyTorch] Fuse eager EP prepare and dispatch into a single op
#3380 opened Aug 13, 2026 by phu0ngng Collaborator • Draft
13 tasks
2
[Pytorch] Optimize CPU overhead for NCCL EP eager mode
#3378 opened Aug 13, 2026 by YangFei1990 Collaborator Loading…
13 tasks
2
fix: duplicate JUnit XML filename overwrites encoder test results community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3376 opened Aug 13, 2026 by andrewwhitecdw Contributor Loading…
2
fix: SBHD reorder skip uses original shape instead of swapped tensor community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3373 opened Aug 13, 2026 by andrewwhitecdw Contributor Loading…
3
fix: comment typo 'mas' -> 'mask' in two TODOs community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3372 opened Aug 13, 2026 by andrewwhitecdw Contributor Loading…
2
fix: make_mask uses deprecated jnp.bool instead of jnp.bool_ community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3371 opened Aug 13, 2026 by andrewwhitecdw Contributor Loading…
3
fix: Optional segment position annotations in mask helpers community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3369 opened Aug 13, 2026 by andrewwhitecdw Contributor Loading…
2
fix: Unreachable backend check after earlier backend skip community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3368 opened Aug 13, 2026 by andrewwhitecdw Contributor Loading…
2
fix: remove redundant self-assignment out_ = out_ community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3367 opened Aug 13, 2026 by andrewwhitecdw Contributor Loading…
5
fix: simplify qkv_format checks using a membership tuple community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3365 opened Aug 13, 2026 by andrewwhitecdw Contributor Loading…
1
fix: simplify redundant softmax version condition community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3363 opened Aug 13, 2026 by andrewwhitecdw Contributor Loading…
2
fix: add missing NVTE_BHSD to qkv format to_string community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3362 opened Aug 13, 2026 by andrewwhitecdw Contributor Loading…
5
fix: add missing NVTE_BHSD_BHSD_BHSD to layout to_string community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3361 opened Aug 13, 2026 by andrewwhitecdw Contributor Loading…
2
[PyTorch] Restore FlashAttention 2 head dim support on sm103 community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3360 opened Aug 12, 2026 by kalectory Loading…
8 of 13 tasks
2
[Common/PyTorch] Fused grouped MXFP8 requantization
#3359 opened Aug 12, 2026 by YangFei1990 Collaborator Loading…
1 of 13 tasks
1
[Common] Use wide instructions in SR to reduce issue-bound bottleneck community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3357 opened Aug 12, 2026 by janekb04 Collaborator • Draft
2 of 13 tasks
Fix FlashAttention local version validation
#3356 opened Aug 12, 2026 by vcherepanov-nv Collaborator Loading…
3 of 13 tasks
1
6
[Pytorch][Do not merge] Let MXFP8 dispatch return plain torch tensor
#3355 opened Aug 12, 2026 by YangFei1990 Collaborator Loading…
13 tasks
3
[DRAFT][JAX] Optimize MoE block 2.19
#3354 opened Aug 12, 2026 by jberchtold-nvidia Collaborator • Draft
7 of 13 tasks
5
[PyTorch] Share CUDA graph memory across dynamic CP variants community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3353 opened Aug 12, 2026 by xiaoyao0115 • Draft
[PyTorch] Decline fused grouped MLP when the backward format is not E4M3 community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3352 opened Aug 12, 2026 by wilyan09007 Loading…
5 of 13 tasks
1
1
[PyTorch] Fused GDN attention 2.19
#3351 opened Aug 12, 2026 by ksivaman Member • Draft
9 of 15 tasks
3
[Pytorch] MOE Sequential Block
#3350 opened Aug 11, 2026 by vthumbe1503 Collaborator • Draft
13 tasks
[PyTorch] Fix MXFP8 master weight cast when a rank owns no shard. community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3348 opened Aug 11, 2026 by rapatel Loading…
8 of 13 tasks
@negvet
3
ProTip! Adding no:label will show everything without a label.