Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
88 Open 1,663 Closed
88 Open 1,663 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

[RFC][PyTorch] Implement Selective Activation Checkpointing for LayerNormMLP (SelectiveLayerNormMLP)
#2311 opened Oct 28, 2025 by jaimec00 Loading…
5 of 10 tasks
33
[DRAFT] Jax (separate from PyTorch) Quickstart documentation
#2310 opened Oct 28, 2025 by tdophung Loading…
6 of 11 tasks
18
[JAX] Fix mesh resource requirement when no mesh
#2307 opened Oct 27, 2025 by jberchtold-nvidia Loading…
8 of 13 tasks
2
[JAX] Make test tolerances stricter
#2306 opened Oct 27, 2025 by jberchtold-nvidia Loading…
8 of 13 tasks
2
Docs fix
#2301 opened Oct 24, 2025 by pggPL Loading…
8 of 12 tasks
28
[JAX] Fix bug with pre scale bias
#2300 opened Oct 23, 2025 by pggPL Loading…
7 of 13 tasks
6
Fix runtime lib loading logic
#2297 opened Oct 23, 2025 by ksivaman Loading…
8 of 13 tasks
6
[PyTorch Debug] NVFP4 debug stats support
#2296 opened Oct 23, 2025 by pggPL • Draft
8 of 13 tasks
1
[common] Remove kvpacked and qkvpacked attention functions for every kernel type.
#2287 opened Oct 20, 2025 by pggPL Loading…
8 of 13 tasks
3
Refactor and checkpoint tests
#2284 opened Oct 17, 2025 by pstjohn Loading…
1
[PyTorch] Framework extensions for TE/common cuBlasMp bindings
#2283 opened Oct 17, 2025 by denera • Draft
6 of 13 tasks
@denera
Add failing HF tests
#2280 opened Oct 16, 2025 by pstjohn • Draft
[common] Misc improvements for attention 2.10.0
#2272 opened Oct 15, 2025 by cyanguwa Loading…
8 of 13 tasks
4
[JAX] A2A CP support
#2271 opened Oct 15, 2025 by pggPL Loading…
8 of 13 tasks
7
[Draft][JAX] E2E encoder sanity test with synthetic data
#2269 opened Oct 13, 2025 by jberchtold-nvidia Loading…
8 of 13 tasks
1
[PyTorch debug] Fixes to debug tests failures
#2268 opened Oct 13, 2025 by pggPL Loading…
7 tasks done
2
[Draft][JAX] Add "initialize" XLA stage to remaining TE/JAX primitives
#2260 opened Oct 10, 2025 by jberchtold-nvidia Loading…
8 of 13 tasks
[Common] Split cast/gated kernels by scaling mode
#2248 opened Oct 8, 2025 by Oleg-Goncharov Loading…
4 of 13 tasks
33
[PyTorch] TE FSDP2 Support for FP8/MXFP8
#2245 opened Oct 7, 2025 by vthumbe1503 • Draft
13 tasks
20
[JAX] Improve CollectiveGEMM examples
#2244 opened Oct 7, 2025 by phu0ngng • Draft
13 tasks
[JAX] xla_home logging during JAX build
#2232 opened Oct 3, 2025 by jberchtold-nvidia Loading…
13 tasks
[JAX] Add support for sink attention in JAX
#2225 opened Oct 1, 2025 by pggPL Loading…
8 of 13 tasks
1
21
Enable SWA with CP for THD input format 2.10.0
#2220 opened Sep 30, 2025 by sudhakarsingh27 Loading…
1 of 6 tasks
9
[PyTorch Debug] Custom feature tutorial.
#2216 opened Sep 30, 2025 by pggPL Loading…
8 of 13 tasks
4
UBNEXT with optional add-rms fuse
#2212 opened Sep 29, 2025 by nv-akorzh Loading…
4
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.