Hub documentation

Libraries

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

Libraries

The Hub has support for dozens of libraries in the Open Source ecosystem. Thanks to the huggingface_hub Python library, it’s easy to enable sharing your models on the Hub. The Hub supports many libraries, and we’re working on expanding this support. We’re happy to welcome to the Hub a set of Open Source libraries that are pushing Machine Learning forward.

The table below summarizes the supported libraries and their level of integration. Find all our supported libraries in the model-libraries.ts file.

Library Description Inference API Widgets Download from Hub Push to Hub
Adapters A unified Transformers add-on for parameter-efficient and modular fine-tuning. βœ… βœ… βœ… βœ…
AllenNLP An open-source NLP research library, built on PyTorch. βœ… βœ… βœ… ❌
Asteroid PyTorch-based audio source separation toolkit βœ… βœ… βœ… ❌
BERTopic BERTopic is a topic modeling library for text and images βœ… βœ… βœ… βœ…
Diffusers A modular toolbox for inference and training of diffusion models βœ… βœ… βœ… βœ…
docTR Models and datasets for OCR-related tasks in PyTorch & TensorFlow βœ… βœ… βœ… ❌
ESPnet End-to-end speech processing toolkit (e.g. TTS) βœ… βœ… βœ… ❌
fastai Library to train fast and accurate models with state-of-the-art outputs. βœ… βœ… βœ… βœ…
Keras Open-source multi-backend deep learning framework, with support for JAX, TensorFlow, and PyTorch. ❌ ❌ βœ… βœ…
KerasNLP Natural language processing library built on top of Keras that works natively with TensorFlow, JAX, or PyTorch. ❌ ❌ βœ… βœ…
TF-Keras (legacy) Legacy library that uses a consistent and simple API to build models leveraging TensorFlow and its ecosystem. ❌ ❌ βœ… βœ…
Flair Very simple framework for state-of-the-art NLP. βœ… βœ… βœ… βœ…
MBRL-Lib PyTorch implementations of MBRL Algorithms. ❌ ❌ βœ… βœ…
MidiTok Tokenizers for symbolic music / MIDI files. ❌ ❌ βœ… βœ…
ML-Agents Enables games and simulations made with Unity to serve as environments for training intelligent agents. ❌ ❌ βœ… βœ…
MLX Model training and serving framework on Apple silicon made by Apple. ❌ ❌ βœ… βœ…
NeMo Conversational AI toolkit built for researchers βœ… βœ… βœ… ❌
OpenCLIP Library for open-source implementation of OpenAI’s CLIP ❌ ❌ βœ… βœ…
PaddleNLP Easy-to-use and powerful NLP library built on PaddlePaddle βœ… βœ… βœ… βœ…
PEFT Cutting-edge Parameter Efficient Fine-tuning Library βœ… βœ… βœ… βœ…
Pyannote Neural building blocks for speaker diarization. ❌ ❌ βœ… ❌
PyCTCDecode Language model supported CTC decoding for speech recognition ❌ ❌ βœ… ❌
Pythae Unified framework for Generative Autoencoders in Python ❌ ❌ βœ… βœ…
RL-Baselines3-Zoo Training framework for Reinforcement Learning, using Stable Baselines3. ❌ βœ… βœ… βœ…
Sample Factory Codebase for high throughput asynchronous reinforcement learning. ❌ βœ… βœ… βœ…
Sentence Transformers Compute dense vector representations for sentences, paragraphs, and images. βœ… βœ… βœ… βœ…
SetFit Efficient few-shot text classification with Sentence Transformers βœ… βœ… βœ… βœ…
spaCy Advanced Natural Language Processing in Python and Cython. βœ… βœ… βœ… βœ…
SpanMarker Familiar, simple and state-of-the-art Named Entity Recognition. βœ… βœ… βœ… βœ…
Scikit Learn (using skops) Machine Learning in Python. βœ… βœ… βœ… βœ…
Speechbrain A PyTorch Powered Speech Toolkit. βœ… βœ… βœ… ❌
Stable-Baselines3 Set of reliable implementations of deep reinforcement learning algorithms in PyTorch ❌ βœ… βœ… βœ…
TensorFlowTTS Real-time state-of-the-art speech synthesis architectures. ❌ ❌ βœ… ❌
Timm Collection of image models, scripts, pretrained weights, etc. βœ… βœ… βœ… βœ…
Transformers State-of-the-art Natural Language Processing for PyTorch, TensorFlow, and JAX βœ… βœ… βœ… βœ…
Transformers.js State-of-the-art Machine Learning for the web. Run πŸ€— Transformers directly in your browser, with no need for a server! ❌ ❌ βœ… ❌
Unity Sentis Inference engine for the Unity 3D game engine ❌ ❌ ❌ ❌

How can I add a new library to the Inference API?

If you’re interested in adding your library, please reach out to us! Read about it in Adding a Library Guide.

< > Update on GitHub