Hugging Face Hub Libraries

Do you only support πŸ€— Transformers?

No, the Hub supports other 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 how they are integrated. 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 βœ… βœ… βœ… βœ…
Adapter Transformers Extends πŸ€—Transformers with Adapters. ❌ ❌ βœ… βœ…
AllenNLP An open-source NLP research library, built on PyTorch. βœ… βœ… βœ… ❌
Asteroid Pytorch-based audio source separation toolkit βœ… βœ… βœ… ❌
ESPnet End-to-end speech processing toolkit (e.g. TTS) βœ… βœ… βœ… ❌
Flair Very simple framework for state-of-the-art NLP. βœ… βœ… βœ… ❌
Pyannote Neural building blocks for speaker diarization. ❌ ❌ βœ… ❌
Sentence Transformers Compute dense vector representations for sentences, paragraphs, and images. βœ… βœ… βœ… βœ…
spaCy Advanced Natural Language Processing in Python and Cython. βœ… βœ… βœ… βœ…
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?

Read about it in Adding a Library Guide.