pytorch
There are 41401 repositories under pytorch topic.
AUTOMATIC1111/stable-diffusion-webui
Stable Diffusion web UI
huggingface/transformers
๐ค 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.
comfyanonymous/ComfyUI
The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface.
rasbt/LLMs-from-scratch
Implement a ChatGPT-like LLM in PyTorch from scratch, step by step
labmlai/annotated_deep_learning_paper_implementations
๐งโ๐ซ 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, ... ๐ง
keras-team/keras
Deep Learning for humans
vllm-project/vllm
A high-throughput and memory-efficient inference and serving engine for LLMs
CorentinJ/Real-Time-Voice-Cloning
Clone a voice in 5 seconds to generate arbitrary speech in real-time
ultralytics/yolov5
YOLOv5 ๐ in PyTorch > ONNX > CoreML > TFLite
ultralytics/ultralytics
Ultralytics YOLO ๐
GokuMohandas/Made-With-ML
Learn how to design, develop, deploy and iterate on production-grade ML applications.
coqui-ai/TTS
๐ธ๐ฌ - a deep learning toolkit for Text-to-Speech, battle-tested in research and production
deepspeedai/DeepSpeed
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
ray-project/ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.
TencentARC/GFPGAN
GFPGAN aims at developing Practical Algorithms for Real-world Face Restoration.
babysor/MockingBird
๐AIๆๅฃฐ: 5็งๅ ๅ ้ๆจ็ๅฃฐ้ณๅนถ็ๆไปปๆ่ฏญ้ณๅ ๅฎน Clone a voice in 5 seconds to generate arbitrary speech in real-time
roboflow/supervision
We write your reusable computer vision tools. ๐
huggingface/pytorch-image-models
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
xinntao/Real-ESRGAN
Real-ESRGAN aims at developing Practical Algorithms for General Image/Video Restoration.
open-mmlab/mmdetection
OpenMMLab Detection Toolbox and Benchmark
facebookresearch/fairseq
Facebook AI Research Sequence-to-Sequence Toolkit written in Python.
yunjey/pytorch-tutorial
PyTorch Tutorial for Deep Learning Researchers
lutzroeder/netron
Visualizer for neural network, deep learning and machine learning models
huggingface/diffusers
๐ค Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.
Lightning-AI/pytorch-lightning
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
JaidedAI/EasyOCR
Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.
svc-develop-team/so-vits-svc
SoftVC VITS Singing Voice Conversion
fastai/fastai
The fastai deep learning library
d2l-ai/d2l-en
Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.
deepinsight/insightface
State-of-the-art 2D and 3D Face Analysis Project
WZMIAOMIAO/deep-learning-for-image-processing
deep learning for image processing including classification and object-detection etc.
junyanz/pytorch-CycleGAN-and-pix2pix
Image-to-Image Translation in PyTorch
microsoft/JARVIS
JARVIS, a system to connect LLMs with ML community. Paper: https://arxiv.org/pdf/2303.17580.pdf
deepset-ai/haystack
AI orchestration framework to build customizable, production-ready LLM applications. Connect components (models, vector DBs, file converters) to pipelines or agents that can interact with your data. With advanced retrieval methods, it's best suited for building RAG, question answering, semantic search or conversational agent chatbots.
pyg-team/pytorch_geometric
Graph Neural Network Library for PyTorch
lukasmasuch/best-of-ml-python
๐ A ranked list of awesome machine learning Python libraries. Updated weekly.