Python Pytorch

Open-source Python projects categorized as Pytorch

Top 23 Python Pytorch Projects

  • GitHub repo transformers

    🤗Transformers: State-of-the-art Natural Language Processing for Pytorch, TensorFlow, and JAX.

    Project mention: [eclectus] - a free tool for stock research, I used nlp to summarize important sec filings | | 2021-06-13

    If I was to do it again I would use Pegasus implemented with hugging face's tranformers

  • GitHub repo Real-Time-Voice-Cloning

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

    Project mention: I decided to create a version of Fitter Happier, but with Thom's synthesized voice. | | 2021-05-29

    I used this github repository and this interview (I chose that interview because the background noise is clean)

  • GitHub repo pytorch-tutorial

    PyTorch Tutorial for Deep Learning Researchers

    Project mention: How to 'practice' pytorch after finishing its basic tutorial? | | 2021-05-09

    I tried to move straight to practicing implementing papers and trying to understand other people's codes but failed miserably. I feel like there was too much of a gap between the basic tutorial and being able to implement ideas into code....hence the question: Is there any resource/way to practice pytorch in general? I did find this and this, but I just wanted to hear what others have gone through to become better at PyTorch up to the point they can build stuff from their own ideas

  • GitHub repo mmdetection

    OpenMMLab Detection Toolbox and Benchmark

    Project mention: [D] Are there any good 3rd party image segmentation/ object detection frameworks besides Tensorflow and Pytorch? | | 2021-03-16

    At least with Pytorch, it is trivial to Frankenstein things and just borrow whatever components you want from other repos. If you do want to look at a framework, I think MMDet is probably the only one that is actually used by some researchers to publish their own work:

  • GitHub repo pytorch-CycleGAN-and-pix2pix

    Image-to-Image Translation in PyTorch

    Project mention: This Wojak Does Not Exist | | 2020-12-31

  • GitHub repo pytorch-lightning

    The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.

    Project mention: [P] An introduction to PyKale​, a PyTorch library that provides a unified pipeline-based API for knowledge-aware multimodal learning and transfer learning on graphs, images, texts, and videos to accelerate interdisciplinary research. Welcome feedback/contribution! | | 2021-04-25

    If you want a good example for reference, take a look at Pytorch Lightning's readme ( It answers the 3 questions of "what is this", "why should I care", and "how do i use it" almost instantly

  • GitHub repo fairseq

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

    Project mention: Wav2vec Unsupervised (wav2vec-U) is a framework for building speech recognition | | 2021-05-21
  • GitHub repo yolov5

    YOLOv5 in PyTorch > ONNX > CoreML > TFLite

    Project mention: Tested YOLOv5 on a football game and it does pretty well. Still, the ball sometimes is not recognized. It would be great to distinguish players as a next goal to this use case. | | 2021-04-28

    I used ultralytics code:

  • GitHub repo EasyOCR

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

    Project mention: Best OCR for keras CNN trained handwritten character model? | | 2021-06-03

    There are many possible approaches to OCR. You might want to take a look at the pipeline of EasyOCR at

  • GitHub repo horovod

    Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

    Project mention: SKLean, TensorFlow, etc vs Spark ML? | | 2021-02-12

    I'm the maintainer for an open source project called Horovod that allows you to distribute deep learning training (e.g., TensorFlow) on platforms like Spark.

  • GitHub repo flair

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

    Project mention: Advice for how to approach classifying apartment posts on facebook? | | 2021-06-04

    For example, my first approach to the pet sentences would be to label all sentences within a respective text corpus containing according information for either yes or no. You would then convert this to a tertiary tag set, something like ["pet allowed", "pet not allowed", "irrelevant"]. You could then try out a model based on SentenceBert, other sentence-level embeddings/language models or 1D CNNs for this. flairNLP ( is a small, little framework which provides comfortable high-level access to different common language models which integrates perfectly with pyTorch.

  • GitHub repo allennlp

    An open-source NLP research library, built on PyTorch.

    Project mention: C4 dataset released (800GB Common Crawl-derived text; T5 training data) | | 2021-03-16
  • GitHub repo d2l-en

    Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 175 universities.

    Project mention: I created a way to learn machine learning through Jupyter | | 2021-04-30

    There are actually some online books and courses built on Jupyter Notebook ([Dive to Deep Learning Book]( for example). However yours is more detail and could really helps beginners.

  • GitHub repo nni

    An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

    Project mention: [D] Efficient ways of choosing number of layers/neurons in a neural network | | 2021-04-20

    optuna, hyperopt, nni, plenty of less-known tools too.

  • GitHub repo Bringing-Old-Photos-Back-to-Life

    Bringing Old Photo Back to Life (CVPR 2020 oral)

    Project mention: Detroit, Michigan (1921) - 1080p Colorized and AI Enhanced [00:07:34] | | 2021-05-07
  • GitHub repo datasets

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

    Project mention: Build an Embeddings index with Hugging Face Datasets | | 2021-01-28

    This article shows how txtai can index and search with Hugging Face's Datasets library. Datasets opens access to a large and growing list of publicly available datasets. Datasets has functionality to select, transform and filter data stored in each dataset.

  • GitHub repo PySyft

    A library for answering questions using data you cannot see

    Project mention: Google Strikes Deal With Hospital Chain to Develop Healthcare Algorithms | | 2021-05-26

    I think this is how it will be done. Look up PySift for how we can extract high-level insights from private datasets while preserving granular privacy.

  • GitHub repo Stanza

    Official Stanford NLP Python Library for Many Human Languages

  • GitHub repo attention-is-all-you-need-pytorch

    A PyTorch implementation of the Transformer model in "Attention is All You Need".

    Project mention: Lack of activation in transformer feedforward layer? | | 2021-05-20

    I'm curious as to why the second matrix multiplication is not followed by an activation unlike the first one. Is there any particular reason why a non-linearity would be trivial or even avoided in the second operation? For reference, variations of this can be witnessed in a number of different implementations, including BERT-pytorch and attention-is-all-you-need-pytorch.

  • GitHub repo MMdnn

    MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Convert models between Caffe, Keras, MXNet, Tensorflow, CNTK, PyTorch Onnx and CoreML. (by microsoft)

    Project mention: [D] Tools for converting TF code to Pytorch | | 2021-02-01
  • GitHub repo best-of-ml-python

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

    Project mention: Are there any speech recognition modules so I can write one and do not have to rely on google and the likes? | | 2021-04-18
  • GitHub repo pix2pixHD

    Synthesizing and manipulating 2048x1024 images with conditional GANs

    Project mention: How to access a class object when I use torch.nn.DataParallel()? | | 2021-03-12

    I used Pix2PixHD implementation in GitHub if you want to see the full code.

  • GitHub repo AlphaPose

    Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System

    Project mention: Human Pose Estimation Recommendation | | 2021-05-02


NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020). The latest post mention was on 2021-06-13.


What are some of the best open-source Pytorch projects in Python? This list will help you:

Project Stars
1 transformers 46,980
2 Real-Time-Voice-Cloning 24,744
3 pytorch-tutorial 20,702
4 mmdetection 15,268
5 pytorch-CycleGAN-and-pix2pix 15,177
6 pytorch-lightning 13,792
7 fairseq 12,942
8 yolov5 12,038
9 EasyOCR 11,678
10 horovod 11,297
11 flair 10,448
12 allennlp 10,093
13 d2l-en 10,071
14 nni 9,769
15 Bringing-Old-Photos-Back-to-Life 8,533
16 datasets 8,352
17 PySyft 7,271
18 Stanza 5,469
19 attention-is-all-you-need-pytorch 5,431
20 MMdnn 5,354
21 best-of-ml-python 5,300
22 pix2pixHD 5,288
23 AlphaPose 5,199