跳到正文
Hugging Face Blog·· 2 天前AI 评分51

Hugging Face 库原生支持 TorchNeuron,AWS Trainium 上运行模型更简单

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

AI 导读

Hugging Face 宣布 Transformers、Diffusers、Accelerate、PEFT 和 TRL 原生支持 TorchNeuron 后端,AWS Trainium 成为可直接使用的设备,该后端于 10 月 20 日进入公开测试,可用 pip install torch-neuronx 安装。

来源:Hugging Face Blog · huggingface.co