跳到正文
热点事件历史事件

AWS 发布在 Trainium 上用 torchtitan 训练 Qwen3 教程

1 篇报道1 个报道来源12 天前更新

先了解这件事

AI 综述

AWS Machine Learning Blog 发布教程,介绍在单台 trn2.48xlarge(16 颗 Trainium2、64 个逻辑 NeuronCore)上用 torchtitan 训练 Qwen3 的三种配置方法。教程称,torchtitan 可通过 TorchNeuron 把 Trainium 注册为原生 PyTorch 设备,沿用同一套 torchrun 启动模式。 教程要求使用 AWS Deep Learning AMI for Neuron 或 Neuron PyTorch 训练容器,并固定文中列出的版本以复现示例;示例将 Qwen3-30B-A3B 缩减为 4 层、64 个专家,以便在单台 trn2.48xlarge 上运行。序列长度需设为 128×LNC 的倍数,LNC 为 2 时为 256。

AI 根据报道生成 · 12 天前更新

报道时间线

沿着报道,了解事件的不同侧面。

10月22日
  1. AWS Machine Learning Blog
    在 AWS Trainium 上用 torchtitan 训练 Qwen3

    在单台 trn2.48xlarge(16 颗 Trainium2、64 个逻辑 NeuronCore)上,torchtitan 可通过 TorchNeuron 把 Trainium 注册为原生 PyTorch 设备,用同一套 torchrun 启动模式训练 Qwen3。