AWS Machine Learning Blog· Josh Longenecker·· 4 小时前AI 评分42
用 torchtitan 在 AWS Trainium 上训练 Qwen3
Train Qwen3 on AWS Trainium with torchtitan
AI 导读
用 torchtitan 在单个 trn2.48xlarge Trainium 实例上训练 Qwen3 有三种方式:Transformers v5 建模后端、torchtitan 原生模型和 NKI MoE 内核。
来源:AWS Machine Learning Blog · aws.amazon.com