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

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

Ultralytics YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking

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

Ultralytics YOLOv5 in PyTorch for object detection, instance segmentation, classification, training, and export.

We write your reusable computer vision tools. 💜

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

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

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

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

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

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

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

VoxCPM2: Tokenizer-Free TTS for Multilingual Speech Generation, Creative Voice Design, and True-to-Life Cloning