跳到正文
在个人 Reader 中保存进度、批注和阅读笔记进入 Reader →
原文
AWS Machine Learning Blog· Josh Longenecker·· 4 小时前AI 评分32

在 AWS Trainium 上用 torchtitan 训练 Qwen3

Train Qwen3 on AWS Trainium with torchtitan

AI 导读

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

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