Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
124 Open 2,037 Closed
124 Open 2,037 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] Avoid autograd's gradient accumulation in grouped MLP if possible
#2871 opened Apr 13, 2026 by ksivaman Member Loading…
6 of 14 tasks
2
Cute Dsl kernel for Wgrad for Fused MOE Layer
#2869 opened Apr 13, 2026 by vthumbe1503 Collaborator Loading…
13 tasks
6
[JAX] Add debug validation mode for runtime group size alignment
#2867 opened Apr 11, 2026 by jberchtold-nvidia Collaborator • Draft
13 tasks
4
Optimizations for MXFP8/NVFP4 dequantize kernels
#2865 opened Apr 10, 2026 by YigongQin • Draft
2 of 13 tasks
Adds GEMM Profiling Guide to TE
#2863 opened Apr 9, 2026 by jomitchellnv Contributor Loading…
7 tasks
5
[DO NOT MERGE] Test CI
#2862 opened Apr 9, 2026 by cyanguwa Collaborator • Draft
13 tasks
Strip local version labels from package version checks
#2858 opened Apr 8, 2026 by pstjohn Contributor Loading…
@ksivaman
4
Add cpplint and ruff linter to pre-commit and fix lint violations
#2853 opened Apr 8, 2026 by pstjohn Contributor Loading…
@ksivaman
7
Bump transformers from 4.55.0 to 5.0.0rc3 in /docs/examples/te_gemma dependencies Pull requests that update a dependency file python Pull requests that update python code
#2851 opened Apr 8, 2026 by dependabot bot Loading…
@sudhakarsingh27
Bump transformers from 4.57.0 to 5.0.0rc3 in /docs/examples/te_llama dependencies Pull requests that update a dependency file python Pull requests that update python code
#2850 opened Apr 8, 2026 by dependabot bot Loading…
@sudhakarsingh27
Skip activation kernels when tensor size is zero bug Something isn't working
#2848 opened Apr 8, 2026 by timmoon10 Collaborator Loading…
8 of 13 tasks
11
[Common] Multicast Fixes
#2847 opened Apr 8, 2026 by phu0ngng Collaborator • Draft
13 tasks
3
[Core] Report CUDA versions when NVRTC compilation fails enhancement New feature or request
#2842 opened Apr 7, 2026 by timmoon10 Collaborator Loading…
8 of 13 tasks
1
13
[CUDNN] Update frontend to version 1.22 and add cuDNN 9.20 path for SM arch >100
#2838 opened Apr 5, 2026 by zmelumian972 Loading…
2 of 3 tasks
@jberchtold-nvidia
3
Add grouped unswizzle functionality for MXFP8 scaling factors
#2837 opened Apr 5, 2026 by int-smart Contributor Loading…
8 of 13 tasks
@ptrendx
7
[PyTorch] Fix FlashAttention 2 head_dim > 192 on sm103 and other architectures
#2836 opened Apr 4, 2026 by pedramr Loading…
1 task done
@cyanguwa
3
[Pyt][Common] Enabling/Guarding sm120 support (non - attention) 2.15.0
#2833 opened Apr 3, 2026 by KshitijLakhani Collaborator • Draft
4 of 13 tasks
@KshitijLakhani
fix CUDA architectures cmake logic community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#2832 opened Apr 3, 2026 by GaetanLepage Contributor Loading…
2 of 13 tasks
@ptrendx
4
Add capture_time_hooks to make_graphed_callables for non-capturable per-callable hooks
#2831 opened Apr 3, 2026 by buptzyb Contributor Loading…
1 of 13 tasks
@ksivaman
17
Port softmax ops to libtorch stable ABI
#2830 opened Apr 3, 2026 by pstjohn Contributor Loading…
22
Cp thd swa with ag
#2829 opened Apr 3, 2026 by sudhakarsingh27 Collaborator • Draft
13 tasks
3
[Common] Reduced padding kernel compilation time
#2827 opened Apr 2, 2026 by Oleg-Goncharov Collaborator Loading…
5 of 13 tasks
2
fix(CP, MLA): CP works fine with MLA in a2a cp_comm_type community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#2826 opened Apr 2, 2026 by zhujian19891203 Contributor Loading…
5 of 13 tasks
@cyanguwa
2
fix(CP, FA): the conditional logic in the FA version contains a vulnerability when processing the output of Flash Attn forward pass community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#2825 opened Apr 2, 2026 by zhujian19891203 Contributor Loading…
5 of 13 tasks
@cyanguwa
2
Parallel Test Execution to decrease CI run times
#2824 opened Apr 2, 2026 by sudhakarsingh27 Collaborator • Draft
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.