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