跳到正文
Hugging Face Blog·· 7 天前AI 评分44

TorchNeuron:在 AWS Trainium 上运行 HF 模型比以往更轻松

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

AI 导读

TorchNeuron 为 PyTorch 在 AWS Trainium 上提供全新原生后端,支持 eager mode、torch.compile 以及分布式 API 等特性。

来源:Hugging Face Blog · huggingface.co