AWS Trainium 和 Inferentia 文档

🤗 最优神经元

Hugging Face's logo
加入 Hugging Face 社区

并获得增强型文档体验

开始使用

🤗 最优神经元

🤗 最优神经元是 🤗 Transformers 库与 AWS 加速器(包括 AWS TrainiumAWS Inferentia)之间的接口。它提供了一套工具,使您能够轻松地在单加速器和多加速器设置中加载、训练和推理不同的下游任务模型。官方验证模型和任务的列表可在 此处 获得。