跳到正文
原文
Hugging Face Blog·· 22 小时前精选AI 评分73

TorchNeuron 为 AWS Trainium 提供 Hugging Face 原生 PyTorch 后端

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

AI 导读

TorchNeuron 为 AWS Trainium 提供原生 PyTorch 后端,Transformers、Diffusers 和 Kernels 可直接使用它运行模型。

推荐理由

文章对比Optimum Neuron与TorchNeuron的开发体验,提供训练与生成示例,便于评估AWS Trainium工作流的迁移方式。

来源:Hugging Face Blog · huggingface.co