Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
145 Open 2,298 Closed
145 Open 2,298 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] Add per-version FlashAttention env vars (NVTE_FLASH_ATTN_V2… community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3204 opened Jul 11, 2026 by wdykas Contributor Loading…
13 tasks
2
[PyTorch][torch.compile] Support for UnfusedDotProductAttention
#3201 opened Jul 10, 2026 by pggPL Collaborator Loading…
7 of 13 tasks
2
[PyTorch][torch.compile] DotProductAttention: declarative packed QKV/KV inputs
#3200 opened Jul 10, 2026 by pggPL Collaborator Loading…
8 of 13 tasks
7
Developer documentation
#3198 opened Jul 9, 2026 by ptrendx Member • Draft
1 of 13 tasks
Add transformer_engine.test() installation smoke check community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3197 opened Jul 9, 2026 by CodersAcademy006 Loading…
1
5
docs: add datatype and hardware support matrix community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3195 opened Jul 9, 2026 by CodersAcademy006 Loading…
2
1
adam: separate param/grad dtypes in AdamCapturableFunctor community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3192 opened Jul 8, 2026 by rapatel Loading…
5 of 13 tasks
2
[PyTorch][torch.compile] Make get_attention_backend traceable without graph breaks
#3189 opened Jul 8, 2026 by pggPL Collaborator Loading…
8 of 13 tasks
5
NCCL EP zero copy additional support community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3187 opened Jul 7, 2026 by YangFei1990 • Draft
7 tasks done
[Common] Pass cu_seqlens and token-unit ragged offsets directly to cuDNN SDPA fprop 2.18 community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3186 opened Jul 7, 2026 by egilliam-nv Loading…
7 of 13 tasks
7
Optimize single-group grouped MLP quantization community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3184 opened Jul 7, 2026 by sraman-rgb Contributor Loading…
13 tasks
7
Prefer Python CUDA libs over implicit system paths build Build system installation
#3182 opened Jul 7, 2026 by sbhavani Collaborator Loading…
5 of 12 tasks
1
4
grouped linear single param fixes community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3178 opened Jul 6, 2026 by CarlosGomes98 Contributor • Draft
3 of 13 tasks
1
Add Graph-safe NVFP4 CUTLASS Group GEMM to unified entry points community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3175 opened Jul 4, 2026 by cael-ling Contributor Loading…
8 tasks done
16
[PyTorch] Enable fused FP8 block-scaling path in GroupedLinear module 2.18 FP8 MoE
#3171 opened Jul 2, 2026 by denera Collaborator Loading…
8 of 13 tasks
1
@denera
16
Reverse MXFP8 quantization row raster community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3170 opened Jul 2, 2026 by sraman-rgb Contributor Loading…
13 tasks
6
Add fused multi-tensor kernel for 1D blockwise FP8 quantization community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3168 opened Jul 2, 2026 by shangxiaokang • Draft
13 tasks
3
Bump transformers from 4.57.0 to 5.3.0 in /docs/examples/te_llama community-contribution PRs from external contributor outside the core maintainers, representing community-driven work. dependencies Pull requests that update a dependency file python Pull requests that update python code
#3167 opened Jul 2, 2026 by dependabot Bot Loading…
@sudhakarsingh27
Bump transformers from 4.55.0 to 5.3.0 in /docs/examples/te_gemma community-contribution PRs from external contributor outside the core maintainers, representing community-driven work. dependencies Pull requests that update a dependency file python Pull requests that update python code
#3166 opened Jul 2, 2026 by dependabot Bot Loading…
@sudhakarsingh27
Improve readability of dgrad overlap variable community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#3165 opened Jul 2, 2026 by Prachi-kushwaha Loading…
4 of 13 tasks
1
3
[JAX] Add attention tutorials 2.18 documentation Improvements or additions to documentation
#3162 opened Jul 1, 2026 by KshitijLakhani Collaborator Loading…
5 of 13 tasks
@KshitijLakhani
11
[PyTorch] Add optional caller-provided output/grad-input buffers to GroupedLinear and fused grouped MLP
#3161 opened Jul 1, 2026 by phu0ngng Collaborator Loading…
7 of 13 tasks
9
[Common][PyTorch] Add strided batched GEMM in BF16/MXFP8 org-contribution
#3160 opened Jul 1, 2026 by yaox12 Member Loading…
8 of 13 tasks
5
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
21
[PyTorch][torch.compile] Add TensorProto mechanism
#3153 opened Jun 29, 2026 by pggPL Collaborator Loading…
4 of 13 tasks
5
ProTip! no:milestone will show everything without a milestone.