跳到正文
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