PyTorch

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

Deep Learning for humans

🧑‍🏫 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, ... 🧠

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

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

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

Ultralytics YOLO 🚀

Learn how to design, 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.

🚀AI拟声: 5秒内克隆您的声音并生成任意语音内容 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.

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