函式庫

Hugging Face Hub 支援開源生態系中的數十種函式庫。歸功於 huggingface_hub Python 函式庫，您可以輕鬆地在 Hub 上分享您的模型。Hub 支援許多函式庫，且我們正致力於擴充此支援範圍。我們很高興歡迎一系列推動機器學習進步的開源函式庫加入 Hub。

下表總結了受支援的函式庫及其整合程度。您可以在 model-libraries.ts 檔案中找到我們所有支援的函式庫。

程式庫	說明	推論提供商	小工具 (Widgets)	從 Hub 下載	推送到 Hub
適配器	一個用於參數高效（parameter-efficient）與模組化微調的統一 Transformers 附加元件。	✅	✅	✅	✅
AllenNLP	一個建構於 PyTorch 之上的開源 NLP 研究函式庫。	✅	✅	✅	❌
Asteroid	基於 PyTorch 的音訊來源分離工具組	✅	✅	✅	❌
BERTopic	BERTopic 是一個用於文字與影像的主題建模函式庫	✅	✅	✅	✅
Diffusers	用於擴散模型（diffusion models）推論與訓練的模組化工具箱	✅	✅	✅	✅
docTR	用於 PyTorch 與 TensorFlow 中 OCR 相關任務的模型與資料集	✅	✅	✅	❌
ESPnet	端對端語音處理工具組（例如 TTS）	✅	✅	✅	❌
fastai	用於訓練具備最先進輸出結果、既快速又精確模型的函式庫。	✅	✅	✅	✅
Keras	開源多後端深度學習框架，支援 JAX、TensorFlow 與 PyTorch。	❌	❌	✅	✅
KerasNLP	建構於 Keras 之上的自然語言處理函式庫，可與 TensorFlow、JAX 或 PyTorch 原生搭配運作。	❌	❌	✅	✅
TF-Keras (舊版)	使用一致且簡單的 API 來建構模型，並利用 TensorFlow 及其生態系的舊版函式庫。	❌	❌	✅	✅
Flair	用於最先進 NLP 的極簡框架。	✅	✅	✅	✅
MBRL-Lib	MBRL 演算法的 PyTorch 實作。	❌	❌	✅	✅
MidiTok	用於符號音樂 / MIDI 檔案的標記化工具 (Tokenizers)。	❌	❌	✅	✅
ML-Agents	讓使用 Unity 製作的遊戲與模擬器能作為訓練智慧代理（intelligent agents）的環境。	❌	❌	✅	✅
MLX	由 Apple 開發，針對 Apple Silicon 的模型訓練與服務框架。	❌	❌	✅	✅
NeMo	為研究人員建構的對話式 AI 工具組	✅	✅	✅	❌
OpenCLIP	OpenAI CLIP 開源實作的函式庫	❌	❌	✅	✅
PaddleNLP	建構於 PaddlePaddle 之上，易於使用且強大的 NLP 函式庫	✅	✅	✅	✅
PEFT	頂尖的參數高效微調（Parameter Efficient Fine-tuning）函式庫	✅	✅	✅	✅
Pyannote	用於說話者分離（speaker diarization）的神經網路建構組件。	❌	❌	✅	❌
PyCTCDecode	支援語音辨識之語言模型的 CTC 解碼器	❌	❌	✅	❌
Pythae	Python 中生成式自動編碼器（Generative Autoencoders）的統一框架	❌	❌	✅	✅
RL-Baselines3-Zoo	使用 Stable Baselines3 的強化學習訓練框架。	❌	✅	✅	✅
Sample Factory	用於高吞吐量非同步強化學習的程式碼庫。	❌	✅	✅	✅
Sentence Transformers	計算句子、段落與影像的密集向量表示。	✅	✅	✅	✅
SetFit	使用 Sentence Transformers 進行高效的小樣本（few-shot）文字分類	✅	✅	✅	✅
spaCy	Python 與 Cython 中的進階自然語言處理。	✅	✅	✅	✅
SpanMarker	熟悉、簡單且最先進的命名實體識別（Named Entity Recognition）。	✅	✅	✅	✅
Scikit Learn (使用 skops)	Python 中的機器學習。	✅	✅	✅	✅
Speechbrain	一個由 PyTorch 驅動的語音工具組。	✅	✅	✅	❌
Stable-Baselines3	在 PyTorch 中針對深度強化學習演算法的一系列可靠實作	❌	✅	✅	✅
TensorFlowTTS	即時、最先進的語音合成架構。	❌	❌	✅	❌
Timm	影像模型、指令碼、預訓練權重等的集合。	✅	✅	✅	✅
Transformers	適用於 PyTorch、TensorFlow 與 JAX 的最先進自然語言處理	✅	✅	✅	✅
Transformers.js	Web 上最先進的機器學習。直接在瀏覽器中執行 🤗 Transformers，無需伺服器！	❌	❌	✅	❌
Unity Sentis	Unity 3D 遊戲引擎的推論引擎	❌	❌	❌	❌

我該如何新增對新函式庫的支援？

如果您有興趣新增您的函式庫，請聯繫我們！請閱讀新增函式庫指南以了解詳情。

在 GitHub 上更新