forked from NVIDIA/TransformerEngine
-
Notifications
You must be signed in to change notification settings - Fork 1
Pull requests: pggPL/TransformerEngine
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
[PyTorch] Declare Bias and the activations as custom ops
#30
opened Aug 5, 2026 by
pggPL
Owner
Loading…
7 of 13 tasks
[PyTorch] torch.compile for an OperationFuser group holding one operation
#29
opened Aug 5, 2026 by
pggPL
Owner
Loading…
7 of 13 tasks
Carry symbolic shapes through compile custom ops
community-contribution
#26
opened Aug 4, 2026 by
shino16
Loading…
13 tasks
[PyTorch][CI] Fix test issues: connect test files that are not wired into qa
#25
opened Jul 30, 2026 by
pggPL
Owner
Loading…
8 of 13 tasks
[JAX] Fix typo in qkv_layout docstring: seperate -> separate
ai-approved
Reviewed by ai_pr_bot and judged trivially simple
#23
opened Jul 27, 2026 by
pggPL
Owner
Loading…
[PyTorch][torch.compile] Support for DotProductAttention on flash and unfused backends
#22
opened Jul 27, 2026 by
pggPL
Owner
Loading…
9 tasks done
[PyTorch] Centralize TE↔torch binary boundary in a single facade file
#21
opened Jul 17, 2026 by
pggPL
Owner
Loading…
5 of 6 tasks
[PyTorch] torch.compile support for UnfusedDotProductAttention
#19
opened Jul 8, 2026 by
pggPL
Owner
Loading…
7 of 13 tasks
[PyTorch] DotProductAttention: declarative packed qkv/kv inputs
#18
opened Jul 7, 2026 by
pggPL
Owner
Loading…
[common/PyTorch] Fused attention: pass real tensor strides to cuDNN (opt-in, v2 C API)
#17
opened Jul 7, 2026 by
pggPL
Owner
Loading…
[PyTorch] Compile-friendly attention API: pointer-free qkv layout detection
#16
opened Jul 7, 2026 by
pggPL
Owner
Loading…
[PyTorch] Deprecate internal max_seqlen derivation in DotProductAttention
#15
opened Jul 3, 2026 by
pggPL
Owner
Loading…
[PyTorch][torch.compile] Replace TensorProto with make_empty_traceable
community-contribution
#13
opened Jul 2, 2026 by
kshitij12345
Loading…
13 tasks
[PyTorch][torch.compile] Add TensorProto mechanism
#8
opened Jun 6, 2026 by
pggPL
Owner
Loading…
4 of 13 tasks
[PyTorch][torch.compile] Make quantizers opaque value objects
#7
opened Jun 6, 2026 by
pggPL
Owner
Loading…
8 of 13 tasks
ProTip!
Updated in the last three days: updated:>2026-08-03.