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

在 AWS Trainium 上用 torchtitan 训练 Qwen3

Train Qwen3 on AWS Trainium with torchtitan

阅读原文

本站未展示全文,请前往来源网站阅读。

AI 导读

在单台 trn2.48xlarge 上,torchtitan 通过 TorchNeuron 将 Trainium 注册为原生 PyTorch 设备,可用三种方式训练 Qwen3-4B-Instruct-2507:Transformers v5 建模后端、torchtitan 原生 Qwen3 实现,以及挂载 NKI MoE kernel。

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