Skip to content

feat: optional torchembed fused Triton RoPE kernel (~3-4x training speedup)#2280

Open
py-ai-dev wants to merge 2 commits into
Lightning-AI:mainfrom
py-ai-dev:feat/torchembed-fused-rope
Open

feat: optional torchembed fused Triton RoPE kernel (~3-4x training speedup)#2280
py-ai-dev wants to merge 2 commits into
Lightning-AI:mainfrom
py-ai-dev:feat/torchembed-fused-rope

Commits

Commits on Jul 6, 2026