Thư viện

Hub hỗ trợ hàng chục thư viện trong hệ sinh thái Mã nguồn mở. Nhờ thư viện Python huggingface_hub, việc bật tính năng chia sẻ mô hình của bạn lên Hub trở nên dễ dàng. Hub hỗ trợ nhiều thư viện và chúng tôi đang nỗ lực mở rộng phạm vi hỗ trợ này. Chúng tôi rất vui mừng chào đón một loạt các thư viện Mã nguồn mở đang thúc đẩy Machine Learning tiến lên trên Hub.

Bảng dưới đây tóm tắt các thư viện được hỗ trợ và mức độ tích hợp của chúng. Tìm tất cả các thư viện được hỗ trợ của chúng tôi trong tệp model-libraries.ts.

Thư viện	Mô tả	Nhà cung cấp suy luận	Widgets	Tải xuống từ Hub	Đẩy lên Hub
Trình thích ứng	Một tiện ích bổ sung cho Transformers, thống nhất cho việc tinh chỉnh hiệu quả tham số và có tính mô-đun.	✅	✅	✅	✅
AllenNLP	Một thư viện nghiên cứu NLP mã nguồn mở, được xây dựng trên PyTorch.	✅	✅	✅	❌
Asteroid	Bộ công cụ tách nguồn âm thanh dựa trên PyTorch	✅	✅	✅	❌
BERTopic	BERTopic là một thư viện mô hình hóa chủ đề cho văn bản và hình ảnh	✅	✅	✅	✅
Diffusers	Bộ công cụ mô-đun cho suy luận và huấn luyện các mô hình khuếch tán	✅	✅	✅	✅
docTR	Các mô hình và bộ dữ liệu cho các tác vụ liên quan đến OCR trong PyTorch & TensorFlow	✅	✅	✅	❌
ESPnet	Bộ công cụ xử lý giọng nói đầu-cuối (ví dụ: TTS)	✅	✅	✅	❌
fastai	Thư viện để huấn luyện các mô hình nhanh và chính xác với kết quả đầu ra tiên tiến nhất.	✅	✅	✅	✅
Keras	Khung học sâu đa backend mã nguồn mở, hỗ trợ JAX, TensorFlow và PyTorch.	❌	❌	✅	✅
KerasNLP	Thư viện xử lý ngôn ngữ tự nhiên được xây dựng trên nền tảng Keras, hoạt động bản địa với TensorFlow, JAX hoặc PyTorch.	❌	❌	✅	✅
TF-Keras (phiên bản cũ)	Thư viện cũ sử dụng API nhất quán và đơn giản để xây dựng các mô hình tận dụng TensorFlow và hệ sinh thái của nó.	❌	❌	✅	✅
Flair	Khung làm việc rất đơn giản cho NLP tiên tiến.	✅	✅	✅	✅
MBRL-Lib	Các cài đặt PyTorch của các thuật toán MBRL.	❌	❌	✅	✅
MidiTok	Tokenizers cho nhạc ký hiệu / tệp MIDI.	❌	❌	✅	✅
ML-Agents	Cho phép các trò chơi và mô phỏng được tạo bằng Unity đóng vai trò là môi trường để huấn luyện các tác tử thông minh.	❌	❌	✅	✅
MLX	Khung đào tạo và phục vụ mô hình trên Apple silicon do Apple phát triển.	❌	❌	✅	✅
NeMo	Bộ công cụ AI hội thoại được xây dựng dành cho các nhà nghiên cứu	✅	✅	✅	❌
OpenCLIP	Thư viện cho bản triển khai mã nguồn mở của CLIP từ OpenAI	❌	❌	✅	✅
PaddleNLP	Thư viện NLP mạnh mẽ và dễ sử dụng được xây dựng trên PaddlePaddle	✅	✅	✅	✅
PEFT	Thư viện Fine-tuning hiệu quả tham số tiên tiến	✅	✅	✅	✅
Pyannote	Các khối xây dựng neural cho phân tách người nói.	❌	❌	✅	❌
PyCTCDecode	Mô hình ngôn ngữ hỗ trợ giải mã CTC cho nhận dạng giọng nói	❌	❌	✅	❌
Pythae	Khung thống nhất cho Autoencoder sinh trong Python	❌	❌	✅	✅
RL-Baselines3-Zoo	Khung huấn luyện cho Học tăng cường, sử dụng Stable Baselines3.	❌	✅	✅	✅
Sample Factory	Mã nguồn cho học tăng cường bất đồng bộ thông lượng cao.	❌	✅	✅	✅
Sentence Transformers	Tính toán các biểu diễn vector đặc cho câu, đoạn văn và hình ảnh.	✅	✅	✅	✅
SetFit	Phân loại văn bản ít mẫu hiệu quả với Sentence Transformers	✅	✅	✅	✅
spaCy	Xử lý ngôn ngữ tự nhiên nâng cao trong Python và Cython.	✅	✅	✅	✅
SpanMarker	Nhận thực thực thể có tên quen thuộc, đơn giản và tiên tiến nhất.	✅	✅	✅	✅
Scikit Learn (sử dụng skops)	Học máy trong Python.	✅	✅	✅	✅
Speechbrain	Bộ công cụ xử lý giọng nói được hỗ trợ bởi PyTorch.	✅	✅	✅	❌
Stable-Baselines3	Bộ các cài đặt đáng tin cậy của các thuật toán học tăng cường sâu trong PyTorch	❌	✅	✅	✅
TensorFlowTTS	Các kiến trúc tổng hợp giọng nói tiên tiến nhất theo thời gian thực.	❌	❌	✅	❌
Timm	Bộ sưu tập các mô hình ảnh, script, trọng số được huấn luyện trước, v.v.	✅	✅	✅	✅
Transformers	Xử lý ngôn ngữ tự nhiên tiên tiến nhất cho PyTorch, TensorFlow và JAX	✅	✅	✅	✅
Transformers.js	Machine Learning tiên tiến nhất cho web. Chạy 🤗 Transformers trực tiếp trong trình duyệt của bạn, không cần máy chủ!	❌	❌	✅	❌
Unity Sentis	Engine suy luận cho engine trò chơi Unity 3D	❌	❌	❌	❌

Làm thế nào để thêm hỗ trợ cho một thư viện mới?

Nếu bạn quan tâm đến việc thêm thư viện của mình, vui lòng liên hệ với chúng tôi! Đọc thêm về điều này trong Hướng dẫn thêm thư viện.

Update on GitHub