Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
91 Open 1,673 Closed
91 Open 1,673 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

[Core] Fix inconsistent logic in C++ tensor class
#2330 opened Nov 1, 2025 by timmoon10 • Draft
7 of 13 tasks
1
[Common] Added an optimized gated rowwise MXFP8 SwiGLU kernel
#2328 opened Oct 31, 2025 by Oleg-Goncharov Loading…
5 of 13 tasks
10
[Pytorch] change fused cross entropy backward grad to fp32 and reduce one read/…
#2325 opened Oct 31, 2025 by RandMist Loading…
8 of 13 tasks
3
[Common] Persistent MXFP8 kernel
#2323 opened Oct 30, 2025 by Oleg-Goncharov • Draft
13 tasks
[JAX] L1_jax_distributed_test suit with individual executions
#2321 opened Oct 30, 2025 by phu0ngng Loading…
7 of 13 tasks
22
[PyTorch] Implement Selective Activation Checkpointing for LayerNormMLP with checkpoint flag
#2311 opened Oct 28, 2025 by jaimec00 Loading…
7 of 13 tasks
1
79
[JAX] Quickstart documentation
#2310 opened Oct 28, 2025 by tdophung Loading…
6 of 11 tasks
65
[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
34
[JAX] Fix bug with pre scale bias
#2300 opened Oct 23, 2025 by pggPL Loading…
7 of 13 tasks
9
Fix runtime lib loading logic
#2297 opened Oct 23, 2025 by ksivaman Loading…
8 of 13 tasks
16
[PyTorch Debug] NVFP4 debug stats support
#2296 opened Oct 23, 2025 by pggPL • Draft
8 of 13 tasks
3
[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
4
[Draft][JAX] Add "initialize" XLA stage to remaining TE/JAX primitives
#2260 opened Oct 10, 2025 by jberchtold-nvidia Loading…
8 of 13 tasks
[PyTorch] FSDP2 Support for TE
#2245 opened Oct 7, 2025 by vthumbe1503 Loading…
13 tasks
108
[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 2.10.0 attention
#2225 opened Oct 1, 2025 by pggPL Loading…
8 of 13 tasks
1
@pggPL
21
ProTip! What’s not been updated in a month: updated:<2025-10-03.