跳到正文
在个人 Reader 中保存进度、批注和阅读笔记进入 Reader →
原文
Hugging Face Blog·· 22 小时前精选AI 评分60

TorchNeuron 让 HF 模型在 AWS Trainium 上运行更简单

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

AI 导读

Hugging Face 宣布 TorchNeuron 为 PyTorch 提供 AWS Trainium 原生后端,Transformers、Diffusers、Accelerate、PEFT 和 TRL 可直接在 Neuron 设备上运行,无需再经过 Optimum Neuron。

推荐理由

原文给出 TorchNeuron 在 HF 库中的原生接入方式与支持比例变化,读者可据此判断迁移成本。

来源:Hugging Face Blog · huggingface.co