PyTorchCon 2026:Meta 团队讲解 torch.accelerator 与 FlexShard
PyTorch 核心开发者讲了两个新工具,FlexShard 把存储分片和计算分片拆开,跑大模型训练的可以看看
Meta 的 Alban Desmaison 和 Edward Yang 在 #PyTorchCon 2026 演讲中介绍 torch.accelerator 如何简化硬件接入,以及通过后端测试工具在昂贵训练前发现故障。演讲还涵盖 torch_remat 用于激活检查点,FlexShard 则将存储分片与计算分片解耦,提升大规模训练可靠性。
Alban Desmaison @albanDesmaison and Edward Yang @ezyang (@Meta) discuss simpler hardware integration and more reliable large-scale training with PyTorch at #PyTorchCon 2026.
They cover torch.accelerator, backend testing, and tools to catch problems before expensive training runs, alongside torch_remat for activation checkpointing and FlexShard for separating storage sharding from compute sharding.
Watch the keynote livestream: https://t.co/6TtaEkvsHz