Best Open Source pytorch Libraries
A curated list of the most popular GitHub repositories tagged with pytorch. Select any project to visualize its architecture and dive into the codebase using RepoMind's AI engine.
#1AUTOMATIC1111/stable-diffusion-webui
Stable Diffusion web UI
#2huggingface/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.
#3Comfy-Org/ComfyUI
The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface.
#4rasbt/LLMs-from-scratch
Implement a ChatGPT-like LLM in PyTorch from scratch, step by step
#5vllm-project/vllm
A high-throughput and memory-efficient inference and serving engine for LLMs
#6labmlai/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, ... 🧠
#7keras-team/keras
Deep Learning for humans
#8CorentinJ/Real-Time-Voice-Cloning
Clone a voice in 5 seconds to generate arbitrary speech in real-time
#9ultralytics/yolov5
YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
#10ultralytics/ultralytics
Ultralytics YOLO 🚀
#11GokuMohandas/Made-With-ML
Learn how to design, develop, deploy and iterate on production-grade ML applications.
#12coqui-ai/TTS
🐸💬 - a deep learning toolkit for Text-to-Speech, battle-tested in research and production
#13deepspeedai/DeepSpeed
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
#14ray-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.
#15TencentARC/GFPGAN
GFPGAN aims at developing Practical Algorithms for Real-world Face Restoration.
#16babysor/MockingBird
🚀Clone a voice in 5 seconds to generate arbitrary speech in real-time
#17roboflow/supervision
We write your reusable computer vision tools. 💜
#18huggingface/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
#19xinntao/Real-ESRGAN
Real-ESRGAN aims at developing Practical Algorithms for General Image/Video Restoration.
#20huggingface/diffusers
🤗 Diffusers: State-of-the-art diffusion models for image, video, and audio generation in PyTorch.
#21lutzroeder/netron
Visualizer for neural network, deep learning and machine learning models
#22open-mmlab/mmdetection
OpenMMLab Detection Toolbox and Benchmark
#23yunjey/pytorch-tutorial
PyTorch Tutorial for Deep Learning Researchers
#24facebookresearch/fairseq
Facebook AI Research Sequence-to-Sequence Toolkit written in Python.
#25Lightning-AI/pytorch-lightning
Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
#26JaidedAI/EasyOCR
Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.
#27d2l-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.
#28svc-develop-team/so-vits-svc
SoftVC VITS Singing Voice Conversion
#29deepinsight/insightface
State-of-the-art 2D and 3D Face Analysis Project
#30fastai/fastai
The fastai deep learning library
#31WZMIAOMIAO/deep-learning-for-image-processing
deep learning for image processing including classification and object-detection etc.
#32junyanz/pytorch-CycleGAN-and-pix2pix
Image-to-Image Translation in PyTorch
#33microsoft/JARVIS
JARVIS, a system to connect LLMs with ML community. Paper: https://arxiv.org/pdf/2303.17580.pdf
#34deepset-ai/haystack
Open-source AI orchestration framework for building context-engineered, production-ready LLM applications. Design modular pipelines and agent workflows with explicit control over retrieval, routing, memory, and generation. Built for scalable agents, RAG, multimodal applications, semantic search, and conversational systems.
#35Anjok07/ultimatevocalremovergui
GUI for a Vocal Remover that uses Deep Neural Networks.
#36pyg-team/pytorch_geometric
Graph Neural Network Library for PyTorch
#37lukasmasuch/best-of-ml-python
🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.
#38Tencent/ncnn
ncnn is a high-performance neural network inference framework optimized for the mobile platform
#39Sanster/IOPaint
Image inpainting tool powered by SOTA AI Model. Remove any unwanted object, defect, people from your pictures or erase and replace(powered by stable diffusion) any thing on your pictures.
#40zergtant/pytorch-handbook
pytorch handbook是一本开源的书籍，目标是帮助那些希望和使用PyTorch进行深度学习开发和研究的朋友快速入门，其中包含的Pytorch教程全部通过测试保证可以成功运行
#41huggingface/datasets
🤗 The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools
#42huggingface/peft
🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.
#43onnx/onnx
Open standard for machine learning interoperability
#44microsoft/onnxruntime
ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator
#45ShusenTang/Dive-into-DL-PyTorch
本项目将《动手学深度学习》(Dive into Deep Learning)原书中的MXNet实现改为PyTorch实现。
#46AI4Finance-Foundation/FinGPT
FinGPT: Open-Source Financial Large Language Models! Revolutionize 🔥 We release the trained model on HuggingFace.
#47meta-llama/llama-cookbook
Welcome to the Llama Cookbook! This is your go to guide for Building with Llama: Getting started with Inference, Fine-Tuning, RAG. We also show you how to solve end to end problems using Llama model family and using them on various provider services
#48sczhou/CodeFormer
[NeurIPS 2022] Towards Robust Blind Face Restoration with Codebook Lookup Transformer
#49arc53/DocsGPT
Private AI platform for agents, assistants and enterprise search. Built-in Agent Builder, Deep research, Document analysis, Multi-model support, and API connectivity for agents.
#50modelscope/FunASR
A Fundamental End-to-End Speech Recognition Toolkit and Open Source SOTA Pretrained Models, Supporting Speech Recognition, Voice Activity Detection, Text Post-processing etc.