AWS Machine Learning Blog· Josh Longenecker·原文发布· 4 小时前AI 评分43
在 AWS Trainium 上用 torchtitan 训练 Qwen3
Train Qwen3 on AWS Trainium with torchtitan
AI 导读
AWS 介绍在单台 trn2.48xlarge(16 个 Trainium2 加速器、64 个逻辑 NeuronCore)上用 torchtitan 训练 Qwen3 的三种方式:Transformers v5 作为建模后端、torchtitan 原生 Qwen3 实现,以及接入 NKI MoE 内核。
本站提供 AI 导读,完整报道请前往来源网站阅读。
阅读原文 AWS Machine Learning Blog · aws.amazon.com