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