「PyTorch」标签文章

专用加速芯片上的 PyTorch 前端：先问编译路径，再算迁移成本

TorchNeuron 把 PyTorch 模型迁到 Trainium 的入口简化成 .to('neuron')，但“前端不变”应拆成 API、语义、图捕获、工具链四个可分别验证的承诺。迁移前应先确认图捕获边界、算子落点、精度与分布…

为 Trainium 编译模型：TorchNeuron 的图优化实践

TorchNeuron 把 PyTorch 图编译到 Trainium 时，性能差距主要由静态 shape、算子融合和并行切分决定；文中给出 trn1.32xlarge 上的实测编译命令、编译日志判读方法与监控阈值。