Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
73 Open 1,479 Closed
73 Open 1,479 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

[JAX] Fix Flax variable creation when quantizers are created directly from a recipe
#2079 opened Aug 16, 2025 by jberchtold-nvidia Loading…
8 of 13 tasks
1
Update list of authorized CI users
#2078 opened Aug 16, 2025 by timmoon10 Loading…
8 of 14 tasks
1
[JAX] Fix for TE GEMM - Always AllGather RHS non-contracting dims with FSDP axis
#2075 opened Aug 14, 2025 by phu0ngng • Draft
7 of 13 tasks
5
[Common] Add checks to CUDA kernel launch and CUDA API calls
#2074 opened Aug 14, 2025 by yaox12 Loading…
13 tasks
7
[PyTorch] Check if the given recipe is supported in fp8_autocast
#2073 opened Aug 14, 2025 by yaox12 Loading…
1 of 13 tasks
1
8
Bump cuDNN FE to 1.14.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
4
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
12
feat: Add support for multiple quantization modes in the UB communicators
#2043 opened Aug 8, 2025 by djns99 • Draft
3 of 13 tasks
2
[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
32
Ubnext
#2038 opened Aug 6, 2025 by nv-akorzh Loading…
Symmetric Memory Pool
#2037 opened Aug 6, 2025 by wdykas Loading…
13 tasks
2
Add backward RMSNorm+Add fusion
#2028 opened Aug 5, 2025 by janekb04 Loading…
8 of 13 tasks
18
Offloading support for multiple attention layouts
#2024 opened Aug 3, 2025 by sanandaraj5597 Loading…
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
[PyTorch][Mcore] Fix illegal memory access issue while using Mcore async checkpoint with fp8 tensorwise recipe bug Something isn't working
#1956 opened Jul 16, 2025 by zhongbozhu Loading…
13 tasks
2
ProTip! Mix and match filters to narrow down what you’re looking for.