huggingface/transformers

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

What it solves

Transformers provides a centralized model-definition framework that allows users to easily access, use, and train state-of-the-art pretrained models across multiple modalities. It eliminates the need to train models from scratch, reducing compute costs and carbon footprints while providing a unified API to lower the barrier to entry for researchers and developers.

How it works

It acts as a pivot across the AI ecosystem by standardizing model definitions. This ensures compatibility with various training frameworks (such as DeepSpeed, PyTorch-Lightning, and FSDP), inference engines (like vLLM and TGI), and other modeling libraries (such as llama.cpp). Users can interact with models via high-level abstractions like the Pipeline API for quick inference or access model internals for deeper customization.

Who it’s for

Researchers, engineers, and developers who want to implement state-of-the-art machine learning models for text, vision, audio, and video tasks without having to build the architectures from scratch.

Highlights

  • Multi-modal support: Works with text, computer vision, audio, video, and multimodal models.
  • Framework agnostic: Allows moving models between PyTorch, JAX, and TF2.0.
  • Massive library: Access to over 1 million pretrained model checkpoints via the Hugging Face Hub.
  • Simplified API: Offers a high-level Pipeline API that handles preprocessing and output for various tasks.

Related

  • Project
  • Project
  • Project
  • Project
  • Project