Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
111 Open 1,984 Closed
111 Open 1,984 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

adds NVFP4 Fused Adam support
#2797 opened Mar 24, 2026 by jomitchellnv Loading…
2 of 13 tasks
14
[JAX] Add warning if using BSHD and max_segments_per_seq > 1
#2796 opened Mar 24, 2026 by jberchtold-nvidia Loading…
8 of 13 tasks
5
If model parameters are DTensors, optimizer states should also be DTensors.
#2795 opened Mar 24, 2026 by cspades Loading…
1 of 13 tasks
12
[JAX] TE GMM v2 enforcement Env Var
#2794 opened Mar 23, 2026 by jberchtold-nvidia • Draft
13 tasks
3
Avoid CPU offload wait_event for validation
#2793 opened Mar 23, 2026 by vasunvidia Loading…
13 tasks
2
[PyTorch][torch.compile] Add value object support for quantizers
#2792 opened Mar 23, 2026 by pggPL • Draft
1 of 13 tasks
[PyTorch] [torch.compile] Remove module reference from autograd function args
#2791 opened Mar 23, 2026 by pggPL Loading…
8 of 13 tasks
2
[WIP] [Common, PyTorch] Add triton mHC kernels & pytorch operators
#2790 opened Mar 23, 2026 by kainzhong • Draft
13 tasks
Optimize fp8 block scaling Allgather for FSDP2
#2789 opened Mar 23, 2026 by vthumbe1503 Loading…
1 of 13 tasks
19
add high precision init weights to fully_shard example
#2785 opened Mar 20, 2026 by pstjohn • Draft
[Common][JAX] Add CUB TopK MaxPairs interface
#2784 opened Mar 20, 2026 by huanghua1994 Loading…
8 of 13 tasks
5
Optimize naive top-k masking in fused router
#2783 opened Mar 19, 2026 by yosh20004 Loading…
3 of 13 tasks
5
Fused Adam Support for MXFP8 + FSDP2 integration
#2780 opened Mar 18, 2026 by vthumbe1503 • Draft
13 tasks
14
[fused_router][pytorch] Optimize naive topk path and add perf benchmark
#2776 opened Mar 18, 2026 by XiaomingFun233 Loading…
6
add mark_not_offload() interface for cpu_offload_v1
#2770 opened Mar 17, 2026 by lhb8125 Loading…
13 tasks
4
GEMM + Swiglu fused Grouped MLP for MXFP8 2.14.0 MoE
#2769 opened Mar 17, 2026 by ksivaman Loading…
13 tasks
43
[Draft]Support for score_mod and score_mod_bprop in cuDNN's sdpa
#2767 opened Mar 16, 2026 by vcherepanov-nv Loading…
2 of 13 tasks
4
Fix the retrieval of overwrite_main_grad
#2764 opened Mar 16, 2026 by shjwudp • Draft
13 tasks
3
[JAX] MXFP8 Grouped GEMM
#2763 opened Mar 14, 2026 by jberchtold-nvidia • Draft
13 tasks
3
[JAX] MXFP8 Grouped Quantize V2
#2760 opened Mar 13, 2026 by jberchtold-nvidia • Draft
13 tasks
4
[PyTorch] transformer_engine.pytorch.autocast suport inside torch.compile
#2759 opened Mar 13, 2026 by pggPL Loading…
4 of 26 tasks
12
[Common] Improved fused MoE aux loss kernel for large # of experts 2.15.0 MoE
#2758 opened Mar 13, 2026 by denera • Draft
4 of 13 tasks
@denera
[JAX] Grouped GEMM Refactor to use first_dims and last_dims
#2749 opened Mar 10, 2026 by jberchtold-nvidia Loading…
1 of 13 tasks
30
[Common] Persistent Grouped NVFP4 quantization kernel
#2743 opened Mar 6, 2026 by Oleg-Goncharov • Draft
8 of 13 tasks
30
[Common] Persistent Grouped MXFP8 quantization kernel enhancement New feature or request MoE
#2738 opened Mar 5, 2026 by Oleg-Goncharov Loading…
9 of 13 tasks
20
ProTip! Adding no:label will show everything without a label.