Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
69 Open 1,308 Closed
69 Open 1,308 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

Manage dependencies and add missing einops req
#1859 opened Jun 7, 2025 by ksivaman Loading…
7 of 13 tasks
1
[PyTorch] Add support for FP8 current scaling in operation-based API enhancement New feature or request testing Improvements to tests or testing infrastructure
#1858 opened Jun 6, 2025 by timmoon10 Loading…
6 of 14 tasks
1
[JAX] GEMM custom op 2.5.0
#1855 opened Jun 6, 2025 by denera Loading…
6 of 13 tasks
11
pyproject.toml 2.5.0
#1852 opened Jun 5, 2025 by ksivaman Loading…
4 of 13 tasks
4
Add support for Fused Attn MLA head_dim_qk != head_dim_v
#1851 opened Jun 4, 2025 by KshitijLakhani • Draft
13 tasks
@KshitijLakhani
1
Draft: Add support for overlapping wgrad NCCL AG with dgrad GEMM
#1849 opened Jun 4, 2025 by djns99 Loading…
4 of 13 tasks
4
[PyTorch] Inference mode disables initializing quantized weights with column-wise usage 2.5.0 bug Something isn't working enhancement New feature or request
#1847 opened Jun 4, 2025 by timmoon10 Loading…
6 of 13 tasks
1
[JAX] Collective GEMM custom op + primitive + minimal supporting functions jax
#1846 opened Jun 3, 2025 by denera Loading…
5 of 13 tasks
@denera
[JAX] TensorUsage + FP8 GEMM with all layouts handling on BW 2.5.0
#1844 opened Jun 3, 2025 by phu0ngng Loading…
8 of 13 tasks
8
[PyTorch Debug] Fixed the empty tensor bug in statistics computation
#1843 opened Jun 3, 2025 by pggPL Loading…
8 of 13 tasks
2
TE Gemma tutorial attempt#2
#1839 opened Jun 2, 2025 by sudhakarsingh27 • Draft
1 task done
Make quantize_ respect the usages of the quantizer
#1836 opened May 31, 2025 by ptrendx Loading…
13 tasks
9
[PyTorch] Use FP16 tols for distributed tests with TF32 compute
#1831 opened May 28, 2025 by timmoon10 Loading…
6 of 13 tasks
4
[PyTorch] Recipe heuristics for initializing quantized weights
#1827 opened May 27, 2025 by timmoon10 • Draft
6 of 13 tasks
9
Add cuBLASMp-backed GEMM-like API to TE common
#1824 opened May 27, 2025 by mk-61 Loading…
4 of 13 tasks
5
Add support for head_dim > 128 2.5.0
#1797 opened May 18, 2025 by cyanguwa Loading…
9 of 13 tasks
8
[PyTorch][MoE] Reduce CPU Overhead By Fuse Torch Empty Calls performance Performance issues
#1793 opened May 16, 2025 by zhongbozhu Loading…
1 of 13 tasks
@zhongbozhu
14
[common] Added support of FP4 data type
#1779 opened May 13, 2025 by Oleg-Goncharov Loading…
6 of 13 tasks
8
[PyTorch] Update PyTorch FSDP2 test to cover all TE layer types testing Improvements to tests or testing infrastructure
#1777 opened May 12, 2025 by denera Loading…
8 of 13 tasks
@denera
8
linear: clear row-wise weight at the end of forward
#1770 opened May 12, 2025 by kshitij12345 • Draft
1
[PyTorch] Draft of new activation offloading API
#1762 opened May 8, 2025 by pggPL • Draft
13 tasks
cache sequence chunk ids for reordering
#1757 opened May 7, 2025 by xrennvidia • Draft
13 tasks
Zr te doc edits
#1745 opened May 2, 2025 by zredeaux07 Loading…
12 tasks
7
[PyTorch] Refactor activation offloading of quantized tensors.
#1738 opened Apr 30, 2025 by pggPL Loading…
8 of 13 tasks
10
fix: update grad_output quant to avoid redundant work
#1736 opened Apr 30, 2025 by kshitij12345 Loading…
1
3
ProTip! Updated in the last three days: updated:>2025-06-06.