Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
74 Open 1,487 Closed
74 Open 1,487 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

[ TE-JAX ] Expose cp_strategy argument to DPA api 2.7.0
#2090 opened Aug 19, 2025 by kocchop Loading…
7 of 12 tasks
1
[JAX] Error checking for mesh resource and update GemmPrimitive to use global_mesh_resource().fsdp_resource 2.7.0
#2088 opened Aug 19, 2025 by jberchtold-nvidia Loading…
1 of 13 tasks
3
[Draft] FP8 AllGather in FP8 GroupedGEMM
#2086 opened Aug 19, 2025 by mingxu1067 Loading…
2 of 13 tasks
Update NGC version to 25.08
#2085 opened Aug 19, 2025 by phu0ngng Loading…
6 of 13 tasks
1
[PyTorch] Add test for TRT integration + fix for mxfp8 export
#2083 opened Aug 19, 2025 by pggPL Loading…
7 tasks done
1
Update list of authorized CI users testing Improvements to tests or testing infrastructure
#2081 opened Aug 18, 2025 by timmoon10 • Draft
8 of 14 tasks
[Common] Add checks to CUDA kernel launch and CUDA API calls
#2074 opened Aug 14, 2025 by yaox12 Loading…
13 tasks
8
Bump cuDNN FE to 1.14.0 2.7.0
#2072 opened Aug 13, 2025 by vcherepanov-nv Loading…
3 of 8 tasks
5
[PyTorch] ONNX export of FP8 Current Scaling
#2068 opened Aug 12, 2025 by pggPL Loading…
8 of 13 tasks
8
Support communication/gemm overlap for [Wgrad->Dgrad] execution order in the bwd pass.
#2065 opened Aug 12, 2025 by fanshiqing Loading…
3 of 13 tasks
1
Support MLA Context Parallel (CP) exchanging latent KV
#2064 opened Aug 12, 2025 by yuzhongw-nvidia Loading…
13 tasks
Feature fast cast-only mxfp8
#2062 opened Aug 12, 2025 by Jianbing-D Loading…
6 of 13 tasks
6
Support PyPI wheel for cuda13 build Build system
#2057 opened Aug 11, 2025 by ksivaman Loading…
2 of 13 tasks
@timmoon10
3
Add better ordering enforcment to split_overlap_rs gemms.
#2056 opened Aug 11, 2025 by chaseblock Loading…
6 of 13 tasks
[Draft] Add primary weighs fp8 support for mxfp8
#2055 opened Aug 11, 2025 by kunlunl Loading…
13 tasks
[Pytorch] Add Cutlass GroupGEMM Support for fine-grained MoE Model
#2045 opened Aug 8, 2025 by cassiewilliam Loading…
1 of 13 tasks
27
feat: Add support for multiple quantization modes in the UB communicators
#2043 opened Aug 8, 2025 by djns99 • Draft
3 of 13 tasks
7
[JAX] Fix layernorm distributed test sharding and collective assertions
#2041 opened Aug 7, 2025 by jberchtold-nvidia Loading…
8 of 13 tasks
12
[PyTorch] Quantizer as API
#2039 opened Aug 7, 2025 by negvet Loading…
5 of 11 tasks
35
Ubnext
#2038 opened Aug 6, 2025 by nv-akorzh Loading…
Symmetric Memory Pool
#2037 opened Aug 6, 2025 by wdykas Loading…
13 tasks
2
Offloading support for multiple attention layouts
#2024 opened Aug 3, 2025 by sanandaraj5597 Loading…
3
Dropout with 8-bit RNG
#2014 opened Jul 30, 2025 by vasunvidia Loading…
13 tasks
28
[Draft] Add FP8 attention with current scaling
#2012 opened Jul 30, 2025 by cyanguwa Loading…
8 of 13 tasks
[C][PyTorch] NVFP4 forward MXFP8 backward recipe
#1970 opened Jul 18, 2025 by ksivaman • Draft
2 of 13 tasks
16
ProTip! Filter pull requests by the default branch with base:main.