跳到正文
原文
Hugging Face Blog·· 1 天前

Hugging Face 推出 TorchNeuron,让模型在 AWS Trainium 上运行更简单

TorchNeuron: running HF models on AWS Trainium is easier than ever

AI 导读

Hugging Face 发布 TorchNeuron,为 PyTorch 在 AWS Trainium 上提供原生后端,Transformers、Diffusers 和 Kernels 原生支持,10 月 20 日起以公开 beta 开放,可通过 pip install torch-neuronx 安装。

入选记录

入选两次之和 120 ≥ 门槛的两倍 120

信源等级
官方一手,这一级的门槛是 60 分
预筛
通过:介绍在AWS Trainium上运行HF模型的AI后端与用法
推荐理由
原文给出 Trainium 原生后端在 Transformers、Diffusers、TRL 上的支持范围与迁移方式,可据此判断现有训练推理脚本的改动量。

评分由模型按同一份标准独立打两次,满分 100;两次之和达到门槛的两倍才入选。门槛按信源等级设定。

来源:Hugging Face Blog · huggingface.co