Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
128 Open 2,028 Closed
128 Open 2,028 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

Bias Prob Scaling for GroupedLinear and Fused MOE Layers
#2864 opened Apr 9, 2026 by vthumbe1503 Collaborator Loading…
13 tasks
9
Adds GEMM Profiling Guide to TE
#2863 opened Apr 9, 2026 by jomitchellnv Contributor Loading…
7 tasks
4
[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
3
Enforce minimum NCCL version for cuBLASMp
#2857 opened Apr 8, 2026 by vcherepanov-nv Collaborator Loading…
3 of 13 tasks
10
[PyTorch] Relax dimension constraints for using fused grouped MLP
#2856 opened Apr 8, 2026 by ksivaman Member Loading…
5 of 13 tasks
5
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
10
[Common] Multicast Fixes
#2847 opened Apr 8, 2026 by phu0ngng Collaborator • Draft
13 tasks
3
Add Megatron-FSDP E2E integration test to TE CI/CD (L1).
#2845 opened Apr 7, 2026 by cspades Member Loading…
3 of 13 tasks
@timmoon10
10
[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
comm_gemm_test fixes
#2839 opened Apr 6, 2026 by almogsegal Contributor Loading…
1 of 13 tasks
@denera
9
[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
6
[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
Fix JAX extension build with NVTE_UB_WITH_MPI=1
#2835 opened Apr 4, 2026 by GaetanLepage Contributor Loading…
2 of 13 tasks
@timmoon10
3
[Pyt][Common] Enabling/Guarding sm120 support (non - attention)
#2833 opened Apr 3, 2026 by KshitijLakhani Collaborator • Draft
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
ProTip! Exclude everything labeled bug with -label:bug.