All Algorithms implemented in Python

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

Tensors and Dynamic neural networks in Python with strong GPU acceleration

FastAPI framework, high performance, easy to learn, fast to code, ready for production

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

🧑‍🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), ga…

Deepfakes Software For All

The simplest, fastest repository for training/finetuning medium-sized GPTs.

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

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

💫 Industrial-strength Natural Language Processing (NLP) in Python

🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.

OpenMMLab Detection Toolbox and Benchmark

PyTorch Tutorial for Deep Learning Researchers

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

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

Machine Learning From Scratch. Bare bones NumPy implementations of machine learning models and algorithms with a focus on accessibility. Aims to cover everything from linear regression to deep lear…

Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

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.

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities

☁️ Build multimodal AI applications with cloud-native stack

Best Practices on Recommendation Systems

🤗 The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools

State-of-the-Art Text Embeddings

PyTorch implementations of Generative Adversarial Networks.

Train transformer language models with reinforcement learning.

OpenAI Baselines: high-quality implementations of reinforcement learning algorithms