PyTorch推出TorchNeuron原生后端：AWS Trainium加速大模型训练与调试

PyTorch官方博客正式发布TorchNeuron，这是一个面向AWS Trainium芯片的开源原生PyTorch后端。该项目允许开发者继续使用已有的模型代码、库和工具，即可将训练负载迁移至Trainium，减少针对专用AI芯片的适配成本。TorchNeuron支持即时执行模式，便于研究人员快速调试和验证模型，也覆盖语言、视频、音频等前沿模型架构，包括稠密与混合专家Transformer、扩散模型、状态空间模型、世界模型和结构预测模型等。目前TorchNeuron已以Apache 2.0开源许可证进入公开Beta阶段，这意味着开发者可以更自由地评估、集成和二次开发。随着大模型训练对算力需求持续上升，AWS Trainium与PyTorch生态的深度打通，有望为AI基础设施提供更高效、更开放的替代方案，并降低云上训练成本。