Skip to content

Pull requests: NVIDIA/TransformerEngine

New pull request New
128 Open 1,999 Closed
128 Open 1,999 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

[Pyt][Common] Enabling/Guarding sm120 support (non - attention)
#2833 opened Apr 3, 2026 by KshitijLakhani • Draft
13 tasks
@KshitijLakhani
fix CUDA architectures cmake logic
#2832 opened Apr 3, 2026 by GaetanLepage Loading…
2 of 13 tasks
2
Add capture_time_hooks to make_graphed_callables for non-capturable per-callable hooks
#2831 opened Apr 3, 2026 by buptzyb Loading…
1 of 13 tasks
3
Port softmax ops to libtorch stable ABI
#2830 opened Apr 3, 2026 by pstjohn Loading…
10
Cp thd swa with ag
#2829 opened Apr 3, 2026 by sudhakarsingh27 • Draft
13 tasks
[Common] Reduced padding kernel compilation time
#2827 opened Apr 2, 2026 by Oleg-Goncharov Loading…
5 of 13 tasks
1
fix(CP, MLA): CP works fine with MLA in a2a cp_comm_type community-contribution PRs from external contributor outside the core maintainers, representing community-driven work.
#2826 opened Apr 2, 2026 by zhujian19891203 Loading…
5 of 13 tasks
2
fix(CP, FA): the conditional logic in the FA version contains a vulnerability when processing the output of Flash Attn forward pass
#2825 opened Apr 2, 2026 by zhujian19891203 Loading…
5 of 13 tasks
1
Parallel Test Execution to decrease CI run times
#2824 opened Apr 2, 2026 by sudhakarsingh27 • Draft
[Common] Fix fused router for large top-K and expert counts
#2821 opened Apr 1, 2026 by harryzhou2000 Loading…
7 of 13 tasks
15
Refactor Amax Kernel ldmatrix loads, TMA/compute barriers, swizzle_idx
#2820 opened Apr 1, 2026 by cael-ling Loading…
6 of 13 tasks
14
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
9
[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
4
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
[PyTorch] [CI] Capture subprocess stderr in distributed tests for better CI error re…
#2802 opened Mar 25, 2026 by sudhakarsingh27 Loading…
13 tasks
4
[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
ProTip! no:milestone will show everything without a milestone.