Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
129 Open 1,989 Closed
129 Open 1,989 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

[JAX] Fix: Use jitted kernels for generating THD (and BSHD) segment pos
#2823 opened Apr 1, 2026 by KshitijLakhani • Draft
5 of 13 tasks
@KshitijLakhani
1
Remove integration test for Lightning-Thunder testing Improvements to tests or testing infrastructure
#2822 opened Apr 1, 2026 by timmoon10 Loading…
8 of 14 tasks
2
Fix fused router for large top-K and expert counts
#2821 opened Apr 1, 2026 by harryzhou2000 Loading…
7 of 13 tasks
5
Refactor Amax Kernel ldmatrix loads, TMA/compute barriers, swizzle_idx
#2820 opened Apr 1, 2026 by cael-ling Loading…
6 of 13 tasks
6
[PyTorch] Fix bug with PR 2677
#2819 opened Mar 31, 2026 by sudhakarsingh27 Loading…
1 of 13 tasks
10
Comm gemm fixes
#2818 opened Mar 31, 2026 by almogsegal Loading…
13 tasks
2
[Pytorch][Common] Hybrid quantization
#2817 opened Mar 31, 2026 by negvet Loading…
1 of 13 tasks
4
[PyTorch] [torch.compile] Remove internal tensor state from Float8CurrentScalingQuantizer
#2816 opened Mar 31, 2026 by pggPL • Draft
13 tasks
Fix nvshmem build
#2815 opened Mar 31, 2026 by GaetanLepage Loading…
2 of 13 tasks
2
Pass input_output_alias to TritonAutotunedKernelCall
#2814 opened Mar 31, 2026 by tdophung Loading…
5 of 13 tasks
5
Add PyTorch stable ABI extension for libtorch compatibility
#2813 opened Mar 30, 2026 by pstjohn • Draft
6
[PyTorch] [torch.compile] Split linear forward into forward and setup context.
#2811 opened Mar 30, 2026 by pggPL Loading…
8 of 13 tasks
5
Streamline group Hadamard ComputeKernel loads
#2810 opened Mar 29, 2026 by cael-ling Loading…
5 of 13 tasks
7
Single __syncthreads per stage in GroupHadamardAmaxTmaKernel
#2809 opened Mar 29, 2026 by cael-ling Loading…
8 of 13 tasks
3
Precomputed swizzle_idx into group Hadamard ComputeKernel
#2808 opened Mar 29, 2026 by cael-ling Loading…
8 of 13 tasks
6
[PyTorch][Flash Attn] Add fallback import for FA3
#2806 opened Mar 26, 2026 by eattia-nvidia Loading…
7 of 13 tasks
3
[PyT] Fix FSDP2 memory leaks for FP8 weight workspaces and transpose caches
#2805 opened Mar 26, 2026 by pstjohn Loading…
3 tasks done
2
3
[PyT][Test] Add xfailing FSDP2 memory leak detection tests
#2803 opened Mar 25, 2026 by pstjohn Loading…
4 tasks done
4
[PyTorch] [CI] Capture subprocess stderr in distributed tests for better CI error re…
#2802 opened Mar 25, 2026 by sudhakarsingh27 Loading…
13 tasks
2
[JAX] Warmup FFIs with "initialize" stage
#2800 opened Mar 25, 2026 by jberchtold-nvidia Loading…
1 of 13 tasks
2
adds NVFP4 Fused Adam support
#2797 opened Mar 24, 2026 by jomitchellnv Loading…
2 of 13 tasks
14
[FSDP2/Megatron-FSDP/DCP] If model parameters are DTensors, optimizer states should also be DTensors.
#2795 opened Mar 24, 2026 by cspades Loading…
1 of 13 tasks
19
[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
3
[PyTorch][torch.compile] Add value object support for quantizers
#2792 opened Mar 23, 2026 by pggPL • Draft
1 of 13 tasks
ProTip! Add no:assignee to see everything that’s not assigned.