Python Pytorch

Open-source Python projects categorized as Pytorch | Edit details

Top 23 Python Pytorch Projects

  • GitHub repo transformers

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

    Project mention: Export and run other machine learning models | | 2021-10-14

    txtai primarily has support for Hugging Face Transformers and ONNX models. This enables txtai to hook into the rich model framework available in Python, export this functionality via the API to other languages (JavaScript, Java, Go, Rust) and even export and natively load models with ONNX.

  • GitHub repo Real-Time-Voice-Cloning

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

    Project mention: Is there a free (preferably open source) offline invoice creation software for windows 10? | | 2021-09-27

    Optimize your datasets for ML. Goodbye, boilerplate code - the fastest dataset optimization and management tool for computer vision.

  • 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 yolov5

    YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

    Project mention: Identification of similar objects (nuts and bolts) | | 2021-09-29

    Take a look at YoloV5. This is a pretrained NN that you can easily adapt to your problem. Its free.

  • GitHub repo mmdetection

    OpenMMLab Detection Toolbox and Benchmark

    Project mention: [D] Good quality code repos on deep learning | | 2021-08-27


  • GitHub repo pytorch-CycleGAN-and-pix2pix

    Image-to-Image Translation in PyTorch

    Project mention: I made a 3d topographic map based on my recent civ6 game | | 2021-08-03

    pix2pix algorithm is used for translating Civ6Maps to heightmaps. Synthesized terrain was rendered in blender.

  • 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

  • Nanos

    Run Linux Software Faster and Safer than Linux with Unikernels.

  • GitHub repo fairseq

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

    Project mention: BART: Denoising Seq2Seq Pre-training for NLG (explained) | | 2021-10-08
  • 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: How would one go about recognizing a text (date) in an image? | | 2021-09-07

    Try running, which reads all the text from a given image. Then loop over the words read and see if it reads your date properly or if it splits it into multiple words. Even if it does this, you can easily find the year (you might need to do some postprocessing on what the ocr reads, i.e transform l to 1). The nice thing is that you get the bounding boxes of where the tect was read so you can use some sort of postprocessing on bbox locations and words read to find the month amd day (i.e. I read 25 at the same y index as the yesr but the x index is to the left, so its probably the day). Good luck :)

  • GitHub repo pytorch_geometric

    Graph Neural Network Library for PyTorch

    Project mention: [D] Asking for ideas on Matrix of only 0/1 values as inputs and real value outputs | | 2021-08-17

    A pytorch implementation:

  • GitHub repo horovod

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

    Project mention: [D] GPU buying recommendation | | 2021-07-17

    If you just want to run tensorflow or pytorch for a Jupyter notebook, setting the environment shouldn't be difficult. I know that AWS has a marketplace of preconfigured images. However, you can go as advanced as setting up a cluster of gpu-equipped nodes to setup Horovod ( to do distributed machine learning. Yes, there's a learning curve, but you cannot acquire this skillet any other way.

  • GitHub repo jina

    Cloud-native neural search framework for 𝙖𝙣𝙮 kind of data

    Project mention: DAE think it's really cool that we have open-source software tools to build our own search engines? | | 2021-10-11

    if you'd like to skip the tutorial and dive head-on:

  • GitHub repo d2l-en

    Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 200 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 flair

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

    Project mention: Preparing data for training NER models | | 2021-10-11

    Training most of the Named Entity Recognition (NER) models for example Flair usually needs to format data in BOI tagging) scheme as shown below where each sentence is separated by blank line

  • GitHub repo allennlp

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

    Project mention: Any allennlp users in this sub? | | 2021-10-08 looks active

  • 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: Automated Machine Learning (AutoML) - 9 Different Ways with Microsoft AI | | 2021-10-04

    For a complete tutorial, navigate to this Jupyter Notebook:

  • GitHub repo datasets

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

    Project mention: Datasets: A Community Library for Natural Language Processing | | 2021-09-08
  • 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 best-of-ml-python

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

    Project mention: Awesome list of ML | | 2021-09-16
  • 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 Stanza

    Official Stanford NLP Python Library for Many Human Languages

  • GitHub repo DeepSpeed

    DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.

    Project mention: Microsoft and NVIDIA AI Introduces MT-NLG: The Largest and Most Powerful Monolithic Transformer Language NLP Model | | 2021-10-13

    Microsoft and NVIDIA present the Megatron-Turing Natural Language Generation model (MT-NLG), powered by DeepSpeed and Megatron, the largest and robust monolithic transformer language model trained with 530 billion parameters.

  • GitHub repo AlphaPose

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

    Project mention: Help finding an appropriate model for human pose estimation | | 2021-09-29

    alphapose: havent tried it, but looks to run at 16fps (maybe faster?) but is only intended for research not commercial. Training script available.

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-10-14.


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

Project Stars
1 transformers 52,449
2 Real-Time-Voice-Cloning 30,439
3 pytorch-tutorial 21,977
4 yolov5 17,104
5 mmdetection 16,706
6 pytorch-CycleGAN-and-pix2pix 15,969
7 pytorch-lightning 15,654
8 fairseq 14,127
9 EasyOCR 12,810
10 pytorch_geometric 12,723
11 horovod 11,737
12 jina 11,559
13 d2l-en 11,202
14 flair 10,849
15 allennlp 10,545
16 nni 10,368
17 datasets 10,108
18 Bringing-Old-Photos-Back-to-Life 9,205
19 best-of-ml-python 5,815
20 attention-is-all-you-need-pytorch 5,801
21 Stanza 5,726
22 DeepSpeed 5,687
23 AlphaPose 5,520
Find remote jobs at our new job board There are 34 new remote jobs listed recently.
Are you hiring? Post a new remote job listing for free.
Scout APM: A developer's best friend. Try free for 14-days
Scout APM uses tracing logic that ties bottlenecks to source code so you know the exact line of code causing performance issues and can get back to building a great product faster.