AWS Trainium & Inferentia 文档
🤗 Optimum Neuron
加入 Hugging Face 社区
并获得增强的文档体验
开始使用
🤗 Optimum Neuron
🤗 Optimum Neuron 是 🤗 Transformers 库与 AWS 加速器(包括 AWS Trainium 和 AWS Inferentia)之间的接口。它提供了一套工具,可以在单加速器和多加速器设置中为不同的下游任务轻松加载、训练和推理模型。官方验证的模型和任务列表可在此处找到:这里。