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 in 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 here!

Library Description Inference API Widgets Download from Hub Push to Hub
πŸ€— Transformers State-of-the-art Natural Language Processing for Pytorch, TensorFlow, and JAX βœ… βœ… βœ… βœ…
πŸ€— Diffusers A modular toolbox for inference and training of diffusion models ❌ ❌ βœ… βœ…
Adapter Transformers Extends πŸ€—Transformers with Adapters. ❌ ❌ βœ… βœ…
AllenNLP An open-source NLP research library, built on PyTorch. βœ… βœ… βœ… ❌
Asteroid Pytorch-based audio source separation toolkit βœ… βœ… βœ… ❌
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 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. βœ… βœ… βœ… ❌
ML-Agents Enables games and simulations made with Unity to serve as environments for training intelligent agents. ❌ ❌ βœ… βœ…
NeMo Conversational AI toolkit built for researchers βœ… βœ… βœ… ❌
Pyannote Neural building blocks for speaker diarization. ❌ ❌ βœ… ❌
PyCTCDecode Language model supported CTC decoding for speech recognition ❌ ❌ βœ… ❌
Pythae Unifyed framework for Generative Autoencoders in Python ❌ ❌ βœ… βœ…
RL-Baselines3-Zoo Training framework for Reinforcement Learning, using Stable Baselines3. ❌ βœ… βœ… βœ…
Sentence Transformers Compute dense vector representations for sentences, paragraphs, and images. βœ… βœ… βœ… βœ…
spaCy Advanced Natural Language Processing in Python and Cython. βœ… βœ… βœ… βœ…
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. βœ… βœ… βœ… βœ…

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.