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

使用 torchtitan 在 AWS Trainium 上训练 Qwen3

Train Qwen3 on AWS Trainium with torchtitan

AI 导读

AWS 演示了通过 torchtitan 在单台 trn2.48xlarge 上训练 Qwen3 的三种方式:Transformers v5、torchtitan 原生 Qwen3,以及加入 NKI MoE 内核。

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