🤗 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.

🦜🔗 The platform for reliable agents.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Robust Speech Recognition via Large-Scale Weak Supervision

A curated list of awesome Machine Learning frameworks, libraries and software.

Interact with your documents using the power of GPT, 100% privately, no data leaks

Rich is a Python library for rich text and beautiful formatting in the terminal.

LlamaIndex is the leading framework for building LLM-powered agents over your data.

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

What the f*ck Python? 😱

A community-maintained Python framework for creating mathematical animations.

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Python packaging and dependency management made easy

Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

Python Fire is a library for automatically generating command line interfaces (CLIs) from absolutely any Python object.

Graph Neural Network Library for PyTorch

Repository to track the progress in Natural Language Processing (NLP), including the datasets and the current state-of-the-art for the most common NLP tasks.

Best Practices on Recommendation Systems

Datasets, Transforms and Models specific to Computer Vision

PyTorch implementations of Generative Adversarial Networks.

Mamba SSM architecture

🦉 Data Versioning and ML Experiments

Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

Hydra is a framework for elegantly configuring complex applications

Perform data science on data that remains in someone else's server

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

Deep universal probabilistic programming with Python and PyTorch