PyTorchCon 深入优化器内核:QR、特征分解与 Cholesky 如何提速训练
PyTorch · x · 2026-10-22
PyTorch 官方账号转述 PyTorchCon 2026 上的技术分享:更少的训练步数不一定意味着更快的训练——关键在优化器背后的线性代数 kernel 实现。
演讲者围绕 QR 分解、对称特征分解和 Cholesky 分解展开,借鉴 GPU MODE 线性代数竞赛的经验,讲解:
- 顺序依赖如何限制 GPU 执行效率;
- 实现层如何暴露可并行的计算、重叠各独立阶段;
- 如何合并更新以减少数据搬运开销。
对做训练栈优化和 kernel 工程的人是难得的一手材料,完整视频已放出。
「Infra」频道最新
- Meta 2026 年资本开支最高将达 1450 亿美元,翻倍以上 — Polymarket · 2026-10-22
- pi 模型推理引擎 rec 两行代码即可在 M5 Max 本地运行 — _rockt · 2026-10-22
- 安全研究员警告:GitLab 漏洞严重,呼吁用户尽快打补丁 — basedjensen · 2026-10-22
- PyTorch 大会 Demo Theater 13 场演示:Wan 2.2 视频生成提速 53.6 倍 — PyTorch · 2026-10-22
- TorchTitan 大更新:支持 DeepSeek-V4 等 5D 并行,Muon 优化提速 79% — PyTorch · 2026-10-22
- JNTC 将 TGV 玻璃基板铜填充时间从一天缩至分钟,明年量产玻璃芯 — toptickcrypto · 2026-10-22