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

AWS 发布在 Trainium 上训练 Qwen3 教程

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

先了解这件事

AI 综述

AWS 发布教程,介绍在单台 trn2.48xlarge 上,用 torchtitan 通过 TorchNeuron 把 Trainium 注册为原生 PyTorch 设备,以三种方式训练 Qwen3-4B-Instruct-2507:Transformers v5 后端、torchtitan 原生 Qwen3 实现,以及接入 NKI MoE kernel。 按教程操作需使用单台 trn2.48xlarge,并搭配 AWS Deep Learning AMI for Neuron 或 Neuron PyTorch 训练容器;设置 CompileConfig(enable=False) 会以 eager 模式运行示例;序列长度需设为 128×LNC 的倍数,LNC 为 2 时即 256。

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

报道时间线

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

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

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