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

在 AWS Trainium 上用 torchtitan 训练 Qwen3

Train Qwen3 on AWS Trainium with torchtitan

AI 导读

在单台 trn2.48xlarge(16 颗 Trainium2、64 个逻辑 NeuronCore)上,torchtitan 可通过 TorchNeuron 把 Trainium 注册为原生 PyTorch 设备。

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

木叶资讯微信公众号二维码

关注公众号,每天收到精选与决策卡

长按识别二维码