跳到正文
Hugging Face Blog·· 21 小时前AI 评分58

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、Accelerate、PEFT 和 TRL 可直接把 neuron 当作普通设备使用,无需再经过 Optimum Neuron。

来源:Hugging Face Blog · huggingface.co