AWS Trainium & Inferentia 文件
🤗 Optimum Neuron
加入 Hugging Face 社群
並獲得增強的文件體驗
開始使用
🤗 Optimum Neuron
🤗 Optimum Neuron 是 🤗 Transformers 庫與 AWS 加速器(包括 AWS Trainium 和 AWS Inferentia)之間的介面。它提供了一套工具,可以在單加速器和多加速器設定中為不同的下游任務輕鬆載入、訓練和推理模型。官方驗證的模型和任務列表可在此處找到:這裡。