跳到正文
原文
Hugging Face·· 1 天前AI 评分48

TorchNeuron:在 AWS Trainium 上运行 Hugging Face 模型更加轻松

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

AI 导读

PyTorch 插件 TorchNeuron 为 AWS Trainium 提供原生后端,支持 eager 模式、torch.compile 与 FSDP 等原生分布式 API。

来源:Hugging Face · huggingface.co