ライブラリ

Hubはオープンソースエコシステムにおける数十のライブラリをサポートしています。huggingface_hub Pythonライブラリのおかげで、Hub上でモデルを共有する機能を簡単に有効にできます。Hubは多くのライブラリをサポートしており、このサポートの拡大に取り組んでいます。機械学習を前進させるオープンソースライブラリをHubに迎え入れることを嬉しく思っています。

以下の表は、サポートされているライブラリとその統合レベルをまとめました。サポートされているすべてのライブラリは、the model-libraries.ts fileで確認できます。

ライブラリ	説明	推論プロバイダー	ウィジェット	Hubからダウンロード	Hubにプッシュ
アダプター	パラメータ効率とモジュール性を備えたファインチューニングのための統合されたTransformersアドオン。	✅	✅	✅	✅
AllenNLP	PyTorchを基盤として構築された、オープンソースのNLP研究ライブラリです。	✅	✅	✅	❌
Asteroid	PyTorchベースの音声ソース分離ツールキット	✅	✅	✅	❌
BERTopic	BERTopicは、テキストと画像向けのトピックモデリングライブラリです	✅	✅	✅	✅
Diffusers	拡散モデルの推論と学習のためのモジュール化されたツールボックス	✅	✅	✅	✅
docTR	PyTorch & TensorFlow における OCR 関連タスク向けのモデルとデータセット	✅	✅	✅	❌
ESPnet	エンドツーエンドの音声処理ツールキット（例: TTS）	✅	✅	✅	❌
fastai	最先端の出力で、高速かつ高精度なモデルを訓練するためのライブラリ。	✅	✅	✅	✅
Keras	JAX、TensorFlow、PyTorchをサポートする、オープンソースのマルチバックエンド深層学習フレームワーク。	❌	❌	✅	✅
KerasNLP	Kerasの上に構築され、TensorFlow、JAX、またはPyTorchとネイティブで動作する自然言語処理ライブラリ。	❌	❌	✅	✅
TF-Keras（レガシー）	TensorFlowとそのエコシステムを活用してモデルを構築するための、一貫性がありシンプルなAPIを使用するレガシーライブラリ。	❌	❌	✅	✅
Flair	最先端のNLPのための非常にシンプルなフレームワーク。	✅	✅	✅	✅
MBRL-Lib	MBRLアルゴリズムのPyTorch実装。	❌	❌	✅	✅
MidiTok	シンボリック音楽/MIDIファイル用のトークナイザー。	❌	❌	✅	✅
ML-Agents	Unityで制作されたゲームやシミュレーションを、インテリジェントなエージェントの学習環境として機能させることを可能にします。	❌	❌	✅	✅
MLX	Apple によって開発された、Apple silicon 向けのモデル学習およびサービングフレームワーク。	❌	❌	✅	✅
NeMo	研究者向けに構築された会話型AIツールキット	✅	✅	✅	❌
OpenCLIP	OpenAIのCLIPのオープンソース実装ライブラリ	❌	❌	✅	✅
PaddleNLP	PaddlePaddleを基盤とした、使いやすく強力なNLPライブラリ	✅	✅	✅	✅
PEFT	最先端のパラメータ効率型ファインチューニングライブラリ	✅	✅	✅	✅
Pyannote	話者分離のためのニューラルな構成要素。	❌	❌	✅	❌
PyCTCDecode	言語モデルがサポートする音声認識用のCTCデコーディング	❌	❌	✅	❌
Pythae	Pythonにおける生成型オートエンコーダーの統合フレームワーク	❌	❌	✅	✅
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ゲームエンジン用の推論エンジン	❌	❌	❌	❌

新しいライブラリへのサポートを追加するにはどうすればよいですか？

ライブラリを追加したい場合は、ぜひご連絡ください。ライブラリ追加ガイドをご覧ください。

Update on GitHub