Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
70 Open 1,210 Closed
70 Open 1,210 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

[JAX] Expose recipes via get_te_recipe()
#1746 opened May 2, 2025 by phu0ngng Loading…
7 of 13 tasks
5
Zr te doc edits
#1745 opened May 2, 2025 by zredeaux07 Loading…
13 tasks
[C][PyTorch] Move multi tensors kernels from PyTorch extensions to core 2.4.0
#1744 opened May 2, 2025 by ksivaman Loading…
9 of 13 tasks
7
[JAX] Fix partitioning issues in LayerNorm and LayerNormMLP layers
#1743 opened May 1, 2025 by jberchtold-nvidia Loading…
8 of 13 tasks
1
[JAX] Removes unneccessary reshapes for FP8 GEMM
#1740 opened Apr 30, 2025 by phu0ngng Loading…
7 of 13 tasks
2
[PyTorch] Refactor activation offloading of quantized tensors.
#1738 opened Apr 30, 2025 by pggPL Loading…
8 of 13 tasks
1
fix: update grad_output quant to avoid redundant work
#1736 opened Apr 30, 2025 by kshitij12345 Loading…
1
[JAX] Fix failing L2 JAX unit tests
#1735 opened Apr 29, 2025 by jberchtold-nvidia Loading…
8 of 13 tasks
5
correct weight quantizer for grouped_linear/layernorm_linear and layernorm_mlp
#1733 opened Apr 29, 2025 by HuangHunag-MT Loading…
8 of 13 tasks
Draft: [WIP] Support Context Parallel for Multi Latent Attention (MLA)
#1729 opened Apr 29, 2025 by yuzhongw-nvidia • Draft
13 tasks
[JAX] Decouple Recipe and ScalingMode
#1728 opened Apr 29, 2025 by jberchtold-nvidia Loading…
8 of 13 tasks
5
Add variance calculation from FusedAdam optimizer states
#1726 opened Apr 28, 2025 by kwyss-nvidia Loading…
7 of 13 tasks
2
[PyTorch] Disable verbose debug logs in CI bug Something isn't working testing Improvements to tests or testing infrastructure
#1725 opened Apr 28, 2025 by timmoon10 • Draft
6 of 14 tasks
1
Enhance recipe compatibility
#1724 opened Apr 28, 2025 by negvet Loading…
6 of 13 tasks
12
[JAX] GroupedDense v.2 without dynamic shape
#1721 opened Apr 25, 2025 by phu0ngng • Draft
13 tasks
[PyTorch] Draft of weight offloading + fused wgrad accumulation
#1713 opened Apr 23, 2025 by pggPL • Draft
2 of 13 tasks
MXFP8 support in Userbuffers enhancement New feature or request
#1711 opened Apr 22, 2025 by timmoon10 Loading…
5 of 13 tasks
23
[JAX] Updated: unbalanced CP with THD format
#1709 opened Apr 22, 2025 by huanghua1994 Loading…
8 of 13 tasks
2
[PyTorch] FP8 Subchannel Recipe With FP8 Gather And Configurable Scaling Factor Tensor Swizzling
#1707 opened Apr 21, 2025 by zhongbozhu Loading…
1 of 13 tasks
27
make grouplinear accept the fp8 input
#1706 opened Apr 21, 2025 by Autumn1998 • Draft
15 tasks
1
Dummy PR to test docs
#1698 opened Apr 18, 2025 by ksivaman Loading…
13 tasks
1
Cpu reload double buffer
#1695 opened Apr 17, 2025 by sanandaraj5597 Loading…
7
Added attention offloading
#1691 opened Apr 15, 2025 by sanandaraj5597 Loading…
1
[JAX] Add collective GEMM without compute/communication overlap
#1675 opened Apr 11, 2025 by philipphack Loading…
1 of 6 tasks
8
[QA] Encapsulate functions in test_utils.sh
#1667 opened Apr 10, 2025 by linxiddd Loading…
11 tasks
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.