跳到正文
原文
AWS Machine Learning Blog· Josh Longenecker·· 3 小时前AI 评分59

在 AWS Trainium 上使用 torchtitan 训练 Qwen3 的三种方法

Train Qwen3 on AWS Trainium with torchtitan

AI 导读

AWS 官方教程演示在单台 trn2.48xlarge 上使用 torchtitan 训练 Qwen3,包含 Transformers v5 后端、torchtitan 原生 Qwen3Model 和 NKI MoE 三种配置。

来源:AWS Machine Learning Blog · aws.amazon.com