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 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.
#28deepinsight/insightface
State-of-the-art 2D and 3D Face Analysis Project
#29fastai/fastai
The fastai deep learning library
#30WZMIAOMIAO/deep-learning-for-image-processing
deep learning for image processing including classification and object-detection etc.
#31junyanz/pytorch-CycleGAN-and-pix2pix
Image-to-Image Translation in PyTorch
#32microsoft/onnxruntime
ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator
#33ShusenTang/Dive-into-DL-PyTorch
ๆฌ้กน็ฎๅฐใๅจๆๅญฆๆทฑๅบฆๅญฆไน ใ(Dive into Deep Learning)ๅไนฆไธญ็MXNetๅฎ็ฐๆนไธบPyTorchๅฎ็ฐใ
#34AI4Finance-Foundation/FinGPT
FinGPT: Open-Source Financial Large Language Models! Revolutionize ๐ฅ We release the trained model on HuggingFace.
#35meta-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
#36sczhou/CodeFormer
[NeurIPS 2022] Towards Robust Blind Face Restoration with Codebook Lookup Transformer
#37arc53/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.
#38stas00/ml-engineering
Machine Learning Engineering Open Book
#39mrdbourke/pytorch-deep-learning
Materials for the Learn PyTorch for Deep Learning: Zero to Mastery course.
#40Mikoto10032/DeepLearning
ๆทฑๅบฆๅญฆไน ๅ ฅ้จๆ็จ, ไผ็งๆ็ซ , Deep Learning Tutorial
#41bharathgs/Awesome-pytorch-list
A comprehensive list of pytorch related content on github,such as different models,implementations,helper libraries,tutorials etc.
#42lukas-blecher/LaTeX-OCR
pix2tex: Using a ViT to convert images of equations into LaTeX code.
#43camenduru/stable-diffusion-webui-colab
stable diffusion webui colab
#44microsoft/Bringing-Old-Photos-Back-to-Life
Bringing Old Photo Back to Life (CVPR 2020 oral)
#45cvat-ai/cvat
Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.
#46modelscope/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.
#47graykode/nlp-tutorial
Natural Language Processing Tutorial for Deep Learning Researchers
#48NVIDIA/DeepLearningExamples
State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.
#49NVIDIA/TensorRT-LLM
TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs. TensorRT LLM also contains components to create Python and C++ runtimes that orchestrate the inference execution in a performant way.
#50OpenMined/PySyft
Perform data science on data that remains in someone else's server