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

[pre-commit.ci] auto fixes from pre-commit.com hooks

a964336
Select commit
Loading
Failed to load commit list.
Lit OSS [bot] / tests.yaml / Lit Job (compiler) failed Jul 9, 2026 in 14m 54s

Job Failed

The CI job failed.

Details

Failure reason: Job ci-run-lightning-ai-litgpt-a964336-tests-yaml-lit-job-compiler-6480fef7 failed