跳到正文
热点事件持续更新

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

1 篇报道1 个报道来源2 小时前更新

先了解这件事

AI 综述

AWS 发布教程,介绍在单台 trn2.48xlarge 上用 torchtitan 训练 Qwen3。torchtitan 通过 TorchNeuron 把 Trainium 注册为原生 PyTorch 设备,教程给出三种训练 Qwen3-4B-Instruct-2507 的方式:Transformers v5 后端、torchtitan 原生 Qwen3 实现,以及接入 NKI MoE kernel。 教程要求使用 AWS Deep Learning AMI for Neuron 或 Neuron PyTorch 训练容器,并固定文中列出的版本以复现示例。示例把 Qwen3-30B-A3B 缩减为 4 层、64 个专家,以便在单台 trn2.48xlarge 上运行;序列长度需设为 128×LNC 的倍数,LNC 为 2 时即 256。

AI 根据报道生成 · 1 小时前更新

报道时间线

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

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。

本事件热度走势

还没有足够的连续观测数据,暂不绘制趋势。