Top PyTorch repositories

Repositories with the GitHub topic “pytorch”: 177 repositories with 3.5M stars between them, mostly written in Python, Jupyter Notebook and C++. The highest-rated is Comfy-Org/ComfyUI, #62 on the bestofgithub Elo ladder.

Biggest movers this week

flashinfer-ai/flashinfer

Open on GitHubCuda

FlashInfer: Kernel Library for LLM Serving

  • 1,668+4 Elo, change today
  • 6.9k stars
  • 1.7k forks
git
git clone https://github.com/flashinfer-ai/flashinfer.git
  • attention
  • cuda
  • distributed-inference
  • gpu

vllm-project/semantic-router

Open on GitHubGo

An open, programmable decision layer for models and compute.

  • 1,602+5 Elo, change today
  • 6.4k stars
  • 1.1k forks
git
git clone https://github.com/vllm-project/semantic-router.git
  • ai-gateway
  • guardrails
  • inference
  • kubernetes

higgsfield-ai/higgsfield

Open on GitHubJupyter Notebook

Fault-tolerant, highly scalable GPU orchestration, and a machine learning framework designed for training models with billions to trillions of parameters

  • 1,586+9 Elo, change today
  • 6.3k stars
  • 1.2k forks
  • archived
git
git clone https://github.com/higgsfield-ai/higgsfield.git
  • cluster-management
  • deep-learning
  • distributed
  • llama

pyannote/pyannote-audio

Open on GitHubJupyter Notebook

Neural building blocks for speaker diarization: speech activity detection, speaker change detection, overlapped speech detection, speaker embedding

  • 1,601+9 Elo, change today
  • 10.8k stars
  • 1.1k forks
git
git clone https://github.com/pyannote/pyannote-audio.git
  • overlapped-speech-detection
  • pretrained-models
  • pytorch
  • speaker-change-detection

snakers4/silero-vad

Open on GitHubPython

Silero VAD: pre-trained enterprise-grade Voice Activity Detector

  • 1,637−1 Elo, change today
  • 10.6k stars
  • 893 forks
git
git clone https://github.com/snakers4/silero-vad.git
  • onnx
  • onnx-runtime
  • onnxruntime
  • pytorch

Top 100 on the Elo ladder

  1. #62
    Comfy-Org/ComfyUI

    The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface. The fastest local…

    2,307 Elo +10147kPython
  2. #111
    vllm-project/vllm

    High-throughput LLM serving engine with PagedAttention and an OpenAI-compatible server.

    2,222 Elo −596.9kPython
  3. #135
    huggingface/transformers

    The reference library for pretrained text, vision and audio models: load, fine-tune and run thousands of checkpoints.

    2,186 Elo −5169kPython
  4. #147
    rasbt/LLMs-from-scratch

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

    2,172 Elo +1108kJupyter Notebook
  5. #261
    OpenBMB/VoxCPM

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

    2,087 Elo +2342.5kPython
  6. #365
    ultralytics/ultralytics

    Ultralytics YOLO27, YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image…

    2,031 Elo +763.4kPython
  7. #401
    NandhaKishorM/laya

    Non-autoregressive System 1 decision engine. Typed choice, score and yes/no decisions over any text in a single forward…

    2,014 Elo −1335.5kPython
  8. #578
    1,947 Elo −61165kPython
  9. #732
    roboflow/supervision

    We write your reusable computer vision tools. 💜

    1,910 Elo −2551.6kPython
  10. #831
    Anjok07/ultimatevocalremovergui

    GUI for a Vocal Remover that uses Deep Neural Networks.

    1,888 Elo +1427.1kPython
  11. #836
    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…

    1,888 Elo −2244.2kPython
  12. #862
    xinntao/Real-ESRGAN

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

    1,882 Elo −2037.5kPython
  13. #932
    GokuMohandas/Made-With-ML

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

    1,869 Elo −5250kJupyter Notebook
  14. #1,010
    MakazhanAlpamys/Soup

    Fine-tune LLMs from one YAML. Layer streaming trains an 8B model on a 4 GB laptop GPU.

    1,854 Elo +5210.5kPython
  15. #1,206
    microsoft/onnxruntime

    ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

    1,826 Elo −1622.3kC++
  16. #1,211
    labmlai/annotated_deep_learning_paper_implementations

    🧑‍🏫 60+ Implementations/tutorials of deep learning papers with side-by-side notes 📝; including transformers (original…

    1,826 Elo −10567.7kPython
  17. #1,218
    deepinsight/insightface

    State-of-the-art 2D and 3D Face Analysis Project

    1,825 Elo −3930.2kPython
  18. #1,277
    1,816 Elo −116.4kPython
  19. #1,280
    microsoft/JARVIS

    JARVIS, a system to connect LLMs with ML community. Paper: https://arxiv.org/pdf/2303.17580.pdf

    1,816 Elo +925.6kPython
  20. #1,407
    d2l-ai/d2l-en

    Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70…

    1,796 Elo −2330kPython
  21. #1,442
    vllm-project/vllm-omni

    A framework for efficient model inference with omni-modality models

    1,792 Elo +817.7kPython
  22. #1,459
    huggingface/diffusers

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

    1,789 Elo −7234.9kPython
  23. #1,469
    modelscope/FunASR

    Open-source speech recognition toolkit for training, inference, streaming ASR, VAD, punctuation, speaker diarization…

    1,788 Elo +1021kPython
  24. #1,485
    coqui-ai/TTS

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

    1,786 Elo −1446.2kPython
  25. #1,517
    keras-team/keras

    Deep Learning for humans

    1,782 Elo −5964.4kPython
  26. #1,560
    ultralytics/yolov5

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

    1,777 Elo −3558.2kPython
  27. #1,668
    CorentinJ/Real-Time-Voice-Cloning

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

    1,765 Elo −7660.2kPython
  28. #1,742
    AI4Finance-Foundation/FinGPT

    FinGPT: Open-Source Financial Large Language Models! Revolutionize 🔥 We release the trained model on HuggingFace.

    1,756 Elo +1421.5kJupyter Notebook
  29. #1,771
    huggingface/datasets

    🤗 The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools

    1,752 Elo −422.1kPython
  30. #1,793
    lutzroeder/netron

    Visualizer for neural network, deep learning and machine learning models

    1,750 Elo −3133.6kJavaScript
  31. #1,801
    deepspeedai/DeepSpeed

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

    1,749 Elo −8743.3kPython
  32. #1,803
    mrdbourke/pytorch-deep-learning

    Materials for the Learn PyTorch for Deep Learning: Zero to Mastery course.

    1,749 Elo −5719.4kJupyter Notebook
  33. #1,860
    LMCache/LMCache

    LMCache: Supercharge Your LLM with the Fastest KV Cache Layer

    1,743 Elo +2712.3kPython
  34. #1,865
    NVIDIA/TensorRT-LLM

    TensorRT LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and supports…

    1,742 Elo −1915kPython
  35. #1,872
    onnx/onnx

    Open standard for machine learning interoperability

    1,742 Elo −3321.6kPython
  36. #2,098
    Tencent/ncnn

    ncnn is a high-performance neural network inference framework optimized for the mobile platform

    1,717 Elo −624kC++
  37. #2,131
    kornia/kornia

    🐍 Geometric Computer Vision Library for Spatial AI

    1,713 Elo −2312.3kPython
  38. #2,164
    open-mmlab/mmdetection

    OpenMMLab Detection Toolbox and Benchmark

    1,709 Elo −6133kPython
  39. #2,288
    huggingface/pytorch-image-models

    The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and…

    1,697 Elo −537.2kPython
  40. #2,297
    fastai/fastai

    The fastai deep learning library

    1,696 Elo −6528.2kJupyter Notebook
  41. #2,312
    JaidedAI/EasyOCR

    Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic…

    1,694 Elo −5330.1kPython
  42. #2,314
    DataTalksClub/machine-learning-zoomcamp

    Learn ML engineering for free in 4 months! Register here 👇🏼

    1,694 Elo +214.9kJupyter Notebook
  43. #2,488
    Lightning-AI/pytorch-lightning

    Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

    1,677 Elo −10231.4kPython
  44. #2,497
    huggingface/peft

    🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.

    1,676 Elo −2121.8kPython
  45. #2,519
    WZMIAOMIAO/deep-learning-for-image-processing

    deep learning for image processing including classification and object-detection etc.

    1,674 Elo −8726.4kPython
  46. #2,531
    yunjey/pytorch-tutorial

    PyTorch Tutorial for Deep Learning Researchers

    1,673 Elo −8032.5kPython
  47. #2,539
    stas00/ml-engineering

    Machine Learning Engineering Open Book

    1,672 Elo −119.7kPython
  48. #2,549
    TencentARC/GFPGAN

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

    1,671 Elo −4237.7kPython
  49. #2,593
    flashinfer-ai/flashinfer

    FlashInfer: Kernel Library for LLM Serving

    1,668 Elo +816.9kCuda
  50. #2,810
    tracel-ai/burn

    Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency…

    1,651 Elo −1616.2kRust
  51. #2,910
    pyg-team/pytorch_geometric

    Graph Neural Network Library for PyTorch

    1,644 Elo −7624.1kPython
  52. #2,933
    junyanz/pytorch-CycleGAN-and-pix2pix

    Image-to-Image Translation in PyTorch

    1,642 Elo −8225.2kPython
  53. #2,992
    snakers4/silero-vad

    Silero VAD: pre-trained enterprise-grade Voice Activity Detector

    1,637 Elo +5710.6kPython
  54. #3,054
    isl-org/Open3D

    Open3D: A Modern Library for 3D Data Processing

    1,633 Elo +1114.2kC++
  55. #3,068
    facebookresearch/fairseq

    Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

    1,632 Elo −5632.2kPython
  56. #3,084
    DLR-RM/stable-baselines3

    PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

    1,630 Elo −3014kPython
  57. #3,104
    sczhou/CodeFormer

    [NeurIPS 2022] Towards Robust Blind Face Restoration with Codebook Lookup Transformer

    1,629 Elo −1018.2kPython
  58. #3,116
    lukasmasuch/best-of-ml-python

    🏆 A ranked list of awesome machine learning Python libraries. Updated weekly.

    1,628 Elo −4223.8k
  59. #3,175
    nerfstudio-project/nerfstudio

    A collaboration friendly studio for NeRFs

    1,625 Elo +1512.2kPython
  60. #3,343
    arc53/DocsGPT

    Private AI platform for agents, assistants and enterprise search. Built-in Agent Builder, Deep research, Document…

    1,615 Elo −3818.3kPython
  61. #3,350
    QwenAudio/SenseVoice

    Open-source SenseVoiceSmall model for Mandarin, Cantonese, English, Japanese, and Korean ASR, language ID, emotion…

    1,614 Elo +399.7kC
  62. #3,425
    Mikoto10032/DeepLearning

    深度学习入门教程, 优秀文章, Deep Learning Tutorial

    1,610 Elo −6217.8kJupyter Notebook
  63. #3,469
    meta-llama/llama-cookbook

    Welcome to the Llama Cookbook! This is your go to guide for Building with Llama: Getting started with Inference…

    1,607 Elo −2418.5kJupyter Notebook
  64. #3,473
    zergtant/pytorch-handbook

    pytorch handbook是一本开源的书籍,目标是帮助那些希望和使用PyTorch进行深度学习开发和研究的朋友快速入门,其中包含的Pytorch教程全部通过测试保证可以成功运行

    1,607 Elo −5321.7kJupyter Notebook
  65. #3,495
    babysor/MockingBird

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

    1,605 Elo −5036.8kPython
  66. #3,541
    vllm-project/semantic-router

    An open, programmable decision layer for models and compute.

    1,602 Elo +816.4kGo
  67. #3,545
    kubeflow/kubeflow

    The Cloud Native AI Platform

    1,602 Elo −415.9k
  68. #3,554
    bharathgs/Awesome-pytorch-list

    A comprehensive list of pytorch related content on github,such as different models,implementations,helper…

    1,602 Elo −1816.7k
  69. #3,557
    vwxyzjn/cleanrl

    High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO…

    1,601 Elo +2910.6kPython
  70. #3,568
    ShusenTang/Dive-into-DL-PyTorch

    本项目将《动手学深度学习》(Dive into Deep Learning)原书中的MXNet实现改为PyTorch实现。

    1,601 Elo −8719.5kJupyter Notebook
  71. #3,569
    pyannote/pyannote-audio

    Neural building blocks for speaker diarization: speech activity detection, speaker change detection, overlapped speech…

    1,601 Elo +6010.8kJupyter Notebook
  72. #3,603
    NVlabs/Sana

    SANA & Sol: Efficient Generative Models and Full-Stack Training & Inference Engines for Image, Video, and World Modeling

    1,598 Elo +389.8kPython
  73. #3,785
    higgsfield-ai/higgsfield

    Fault-tolerant, highly scalable GPU orchestration, and a machine learning framework designed for training models with…

    1,586 Elo +746.3kJupyter Notebook
  74. #3,804
    mlfoundations/open_clip

    An open source implementation of CLIP.

    1,585 Elo +1314.2kPython
  75. #3,849
    svc-develop-team/so-vits-svc

    SoftVC VITS Singing Voice Conversion

    1,582 Elo −6128kPython
  76. #3,901
    Sanster/IOPaint

    Image inpainting tool powered by SOTA AI Model. Remove any unwanted object, defect, people from your pictures or erase…

    1,579 Elo −1823.2kPython
  77. #3,940
    QuentinFuxa/WhisperLiveKit

    Real-time, local speech-to-text with streaming ASR, speaker diarization, translation, and OpenAI/Deepgram-compatible…

    1,577 Elo +4311.2kPython
  78. #3,949
    kyegomez/OpenMythos

    A theoretical reconstruction of the Claude Mythos architecture, built from first principles using the available…

    1,576 Elo −2314.9kPython
  79. #4,048
    MITDeepLearning/introtodeeplearning

    Lab Materials for MIT 6.S191: Introduction to Deep Learning

    1,571 Elo +168.8kJupyter Notebook
  80. #4,224
    tangyudi/Ai-Learn

    人工智能学习路线图,整理近200个实战案例与项目,免费提供配套教材,零基础入门,就业实战!包括:Python,数学,机器学习,数据分析,深度学习,计算机视觉,自然语言处理,PyTorch tensorflow…

    1,561 Elo −413.4k
  81. #4,244
    Jack-Cherish/PythonPark

    Python 开源项目之「自学编程之路」,保姆级教程:AI实验室、宝藏视频、数据结构、学习指南、机器学习实战、深度学习实战、网络爬虫、大厂面经、程序人生、资源分享。

    1,559 Elo +1711.9kPython
  82. #4,320
    lukas-blecher/LaTeX-OCR

    pix2tex: Using a ViT to convert images of equations into LaTeX code.

    1,555 Elo −2216.6kPython
  83. #4,329
    speechbrain/speechbrain

    A PyTorch-based Speech Toolkit

    1,555 Elo −311.9kPython
  84. #4,361
    NVIDIA/DeepLearningExamples

    State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and…

    1,553 Elo −1414.8kJupyter Notebook
  85. #4,390
    Megvii-BaseDetection/YOLOX

    YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO…

    1,551 Elo +2510.7kPython
  86. #4,413
    microsoft/LoRA

    Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"

    1,550 Elo −1813.9kPython
  87. #4,548
    Project-MONAI/MONAI

    AI Toolkit for Healthcare Imaging

    1,542 Elo +278.8kPython
  88. #4,611
    milesial/Pytorch-UNet

    PyTorch implementation of the U-Net for image semantic segmentation with high quality images

    1,539 Elo −2911.7kPython
  89. #4,776
    WongKinYiu/yolov7

    Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

    1,530 Elo −5114.1kJupyter Notebook
  90. #4,785
    autogluon/autogluon

    Fast and Accurate ML in 3 Lines of Code

    1,530 Elo +810.8kPython
  91. #4,831
    advimman/lama

    🦙 LaMa Image Inpainting, Resolution-robust Large Mask Inpainting with Fourier Convolutions, WACV 2022

    1,528 Elo +1510.3kJupyter Notebook
  92. #4,881
    open-mmlab/mmpose

    OpenMMLab Pose Estimation Toolbox and Benchmark.

    1,525 Elo +268kPython
  93. #5,058
    unifyai/ivy

    Convert Machine Learning Code Between Frameworks

    1,516 Elo −4914.1kPython
  94. #5,141
    graykode/nlp-tutorial

    Natural Language Processing Tutorial for Deep Learning Researchers

    1,512 Elo −5314.9kJupyter Notebook
  95. #5,153
    bigscience-workshop/petals

    🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading

    1,512 Elo +110.7kPython
  96. #5,183
    chenyuntc/pytorch-book

    PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation…

    1,511 Elo −3512.8kJupyter Notebook
  97. #5,187
    dragen1860/Deep-Learning-with-TensorFlow-book

    深度学习入门开源书,基于TensorFlow 2.0案例实战。Open source Deep Learning book, based on TensorFlow 2.0 framework.

    1,510 Elo −3413.2kJupyter Notebook
  98. #5,193
    qubvel-org/segmentation_models.pytorch

    Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

    1,510 Elo −1411.8kPython
  99. #5,209
    flairNLP/flair

    A very simple framework for state-of-the-art Natural Language Processing (NLP)

    1,509 Elo −3314.4kPython
  100. #5,308
    charliedream1/ai_quant_trade

    Stock AI Trader: 1-stop platform for learning, sim & live trading. Covers: stock basics, strategies, LLMs, factor…

    1,505 Elo +466.7kPython