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