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

[JAX] Fix bug with pre scale bias
#2300 opened Oct 23, 2025 by pggPL Loading…
7 of 13 tasks
[Common] Fix checks in quantize_transpose_vector_blockwise_fp4
#2299 opened Oct 23, 2025 by jberchtold-nvidia Loading…
8 of 13 tasks
Fix runtime lib loading logic
#2297 opened Oct 23, 2025 by ksivaman Loading…
8 of 13 tasks
[PyTorch Debug] NVFP4 debug stats support
#2296 opened Oct 23, 2025 by pggPL Draft
8 of 13 tasks
Remove nvidia-mathdx dependency
#2295 opened Oct 23, 2025 by ksivaman Loading…
6 of 13 tasks
Submodule checkout during setup build Build system enhancement New feature or request
#2293 opened Oct 22, 2025 by ksivaman Loading…
8 of 13 tasks
Refactor and checkpoint tests
#2284 opened Oct 17, 2025 by pstjohn Loading…
Add failing HF tests
#2280 opened Oct 16, 2025 by pstjohn Draft
[common] Misc improvements for attention 2.9.0
#2272 opened Oct 15, 2025 by cyanguwa Loading…
8 of 13 tasks
[JAX] A2A CP support
#2271 opened Oct 15, 2025 by pggPL Loading…
8 of 13 tasks
[Draft][JAX] E2E encoder sanity test with synthetic data
#2269 opened Oct 13, 2025 by jberchtold-nvidia Loading…
8 of 13 tasks
[PyTorch debug] Fixes to debug tests failures
#2268 opened Oct 13, 2025 by pggPL Loading…
7 tasks done
[Common] Split cast/gated kernels by scaling mode
#2248 opened Oct 8, 2025 by Oleg-Goncharov Loading…
4 of 13 tasks
[PyTorch] TE FSDP2 Support for FP8/MXFP8
#2245 opened Oct 7, 2025 by vthumbe1503 Draft
13 tasks
[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
Enable SWA with CP for THD input format
#2220 opened Sep 30, 2025 by sudhakarsingh27 Loading…
1 of 6 tasks
[PyTorch Debug] Custom feature tutorial.
#2216 opened Sep 30, 2025 by pggPL Loading…
8 of 13 tasks
UBNEXT with optional add-rms fuse
#2212 opened Sep 29, 2025 by nv-akorzh Loading…
ProTip! Updated in the last three days: updated:>2025-10-20.