Skip to content

Pull requests: NVIDIA/TransformerEngine

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
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort

Pull requests list

[C][PyTorch] Remove deprecated device_id arg for multi tensor API
#1994 opened Jul 24, 2025 by ksivaman Loading…
6 of 13 tasks
[JAX] Temporary disable Norm custom calls
#1993 opened Jul 24, 2025 by phu0ngng Draft
13 tasks
[PyTorch] Optimize cudagraph static_grad_outputs reuse
#1992 opened Jul 24, 2025 by buptzyb Loading…
5 of 13 tasks
Fixed double buffering issue for assymetric layers
#1984 opened Jul 23, 2025 by sanandaraj5597 Loading…
Use userbuffers for MXFP8 wgrad all-gather overlap
#1982 opened Jul 22, 2025 by djns99 Draft
1 of 13 tasks
Rename do_not_clear to _do_not_clear
#1977 opened Jul 21, 2025 by janekb04 Loading…
3 of 13 tasks
[C][PyTorch] NVFP4 forward MXFP8 backward recipe
#1970 opened Jul 18, 2025 by ksivaman Draft
2 of 13 tasks
[PyTorch] Enable generic QK norm support (+ RMSNorm/LayerNorm)
#1966 opened Jul 18, 2025 by negvet Loading…
7 of 13 tasks
[PyTorch Debug] Debug support for GroupedLinear
#1953 opened Jul 15, 2025 by pggPL Draft
13 tasks
[PyTorch] Refactor C++ quantizer infrastructure
#1952 opened Jul 15, 2025 by timmoon10 Loading…
8 of 13 tasks
[PyTorch] Support delay_wgrad_compute cudagraph
#1948 opened Jul 14, 2025 by buptzyb Loading…
2 of 13 tasks
[Minor] Update 1_getting_started.rst documentation Improvements or additions to documentation
#1947 opened Jul 13, 2025 by dupeljan Loading…
3 of 8 tasks
[JAX] Select cuDNN backend for normalization
#1946 opened Jul 11, 2025 by phu0ngng Loading…
13 tasks
[BUILD] Exclude ninja from required packages
#1932 opened Jul 7, 2025 by phu0ngng Loading…
5 of 13 tasks
Fix import error when flash attention 3 is installed
#1913 opened Jun 30, 2025 by HollowMan6 Loading…
7 of 13 tasks
[PyTorch debug] Improve precision debug tools performance
#1909 opened Jun 30, 2025 by pggPL Loading…
9 of 13 tasks
[PyTorch] Support FA3 MLA CP feature
#1907 opened Jun 28, 2025 by zhujian19891203 Loading…
7 of 13 tasks
[PyTorch Debug] Support log fp8 tensor stats for blockwise recipe
#1905 opened Jun 27, 2025 by lengerfulluse Loading…
12 tasks
[Common] NVFP4 kernels enhancement New feature or request
#1904 opened Jun 27, 2025 by Oleg-Goncharov Draft
5 of 13 tasks
ProTip! Follow long discussions with comments:>50.