Stable Diffusion web UI

🤗 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 most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface.

Implement a ChatGPT-like LLM in PyTorch from scratch, step by step

A high-throughput and memory-efficient inference and serving engine for LLMs

🧑‍🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original, xl, switch, feedback, vit, ...), optimizers (adam, adabelief, sophia, ...), gans(cyclegan, stylegan2, ...), 🎮 reinforcement learning (ppo, dqn), capsnet, distillation, ... 🧠

Deep Learning for humans

Clone a voice in 5 seconds to generate arbitrary speech in real-time

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

Ultralytics YOLO 🚀

Learn how to develop, deploy and iterate on production-grade ML applications.

🐸💬 - a deep learning toolkit for Text-to-Speech, battle-tested in research and production

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

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

GFPGAN aims at developing Practical Algorithms for Real-world Face Restoration.

🚀Clone a voice in 5 seconds to generate arbitrary speech in real-time

We write your reusable computer vision tools. 💜

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

Real-ESRGAN aims at developing Practical Algorithms for General Image/Video Restoration.

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

Visualizer for neural network, deep learning and machine learning models

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.

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

Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.

State-of-the-art 2D and 3D Face Analysis Project

SoftVC VITS Singing Voice Conversion

The fastai deep learning library