Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
127 Open 2,052 Closed
127 Open 2,052 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

Add optimised top-k kernel AIR.
#2890 opened Apr 16, 2026 by dcampora Loading…
8 of 13 tasks
4
[JAX] Fix grouped quant checkpointing
#2889 opened Apr 16, 2026 by jberchtold-nvidia Collaborator Loading…
8 of 13 tasks
4
[PyTorch] Minor optimizations in fused grouped MLP
#2888 opened Apr 15, 2026 by ksivaman Member Loading…
6 of 14 tasks
3
Add AI written qwen3_moe example
#2887 opened Apr 15, 2026 by skyw Loading…
4 of 13 tasks
3
[PyTorch] Add method for mcore to register wgrad accumulation hook
#2886 opened Apr 15, 2026 by ksivaman Member Loading…
7 of 13 tasks
4
Scaled Bias Add support after CUBLAS GGEMM
#2885 opened Apr 15, 2026 by vthumbe1503 Collaborator Loading…
13 tasks
4
[Debug] Add AutoswitchGEmm for Debug Precision Tool
#2883 opened Apr 15, 2026 by shangxiaokang • Draft
3 of 13 tasks
@pggPL
SMEM offset caching RHT
#2882 opened Apr 15, 2026 by sraman-rgb Loading…
13 tasks
7
Check numerics in MXFP8 C++ tests by dequantizing to FP32 2.15.0 testing Improvements to tests or testing infrastructure
#2881 opened Apr 15, 2026 by timmoon10 Collaborator Loading…
6 of 13 tasks
5
fix(readme): update broken links and modernize project description
#2879 opened Apr 14, 2026 by sbhavani Collaborator Loading…
3 of 13 tasks
3
[PyTorch] Split TE ops op_forward into op_forward and setup_context
#2877 opened Apr 14, 2026 by pggPL Collaborator • Draft
13 tasks
8
[DONOT MERGE] Wgrad cute dsl v2
#2872 opened Apr 13, 2026 by vthumbe1503 Collaborator • Draft
13 tasks
2
[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 Loading…
8 of 13 tasks
6
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
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
[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
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.