Repository Index

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

• nlp
• natural-language-processing
• pytorch
• pytorch-transformers

Stable Diffusion web UI

• deep-learning
• diffusion
• image-generation
• image2image

The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface. The fastest local inference engine in the world.

• stable-diffusion
• pytorch
• ai
• python

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

• gpt
• large-language-models
• llm
• python

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

• gpt
• llm
• pytorch
• model-serving

🧑‍🏫 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
• deep-learning-tutorial
• pytorch
• gan

Deep Learning for humans

• deep-learning
• tensorflow
• neural-networks
• machine-learning

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

• ultralytics
• yolov8
• yolo
• deep-learning

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

• deep-learning
• pytorch
• tensorflow
• tts

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

• yolov5
• object-detection
• pytorch
• onnx

We write your reusable computer vision tools. 💜

• computer-vision
• image-processing
• python
• yolo

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

• machine-learning
• deep-learning
• pytorch
• natural-language-processing

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

• python
• text-to-speech
• deep-learning
• speech

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

• ray
• distributed
• parallel
• machine-learning

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

• deep-learning
• pytorch
• gpu
• machine-learning

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

• audio
• deeplearning
• minicpm
• python

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

• pytorch
• gan
• deep-learning
• super-resolution

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

• esrgan
• pytorch
• real-esrgan
• super-resolution

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

• pytorch
• resnet
• pretrained-models
• pretrained-weights

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

• ai
• speech
• pytorch
• deep-learning

Non-autoregressive System 1 decision engine. Typed choice, score and yes/no decisions over any text in a single forward pass, in 100+ languages, with a router that picks the right checkpoint per request.

• calibration
• classification
• decision-model
• huggingface

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

• deep-learning
• diffusion
• image-generation
• pytorch

Visualizer for neural network, deep learning and machine learning models

• neural-network
• deep-learning
• machine-learning
• deeplearning

OpenMMLab Detection Toolbox and Benchmark

• object-detection
• instance-segmentation
• fast-rcnn
• faster-rcnn