Python Pytorch

Open-source Python projects categorized as Pytorch

Top 23 Python Pytorch Projects

  • transformers

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

    Latest mention: Confidence score prediction of huggingface transformers' models for extractive QA task | | 2020-12-27

  • Real-Time-Voice-Cloning

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

    Latest mention: I used AI tools to generate audio of SpongeBob rapping a portion of "Gangster's Paradise" | | 2021-01-05

    I've been struggling to get this to work. But I always get stuck on mis-matching python libraries.

  • pytorch-tutorial

    PyTorch Tutorial for Deep Learning Researchers

    Latest mention: [P] Probabilistic Machine Learning: An Introduction, Kevin Murphy's 2021 e-textbook is out | | 2021-01-01
  • pytorch-CycleGAN-and-pix2pix

    Image-to-Image Translation in PyTorch

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

  • mmdetection

    OpenMMLab Detection Toolbox and Benchmark

    Latest mention: [D] How to organize deep learning projects on Github ? | | 2021-01-10

    You can check these repos:

  • pytorch-lightning

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

    Latest mention: Weekly Developer Roundup #23 - Sun Nov 22 2020 | | 2020-11-21

    PyTorchLightning/pytorch-lightning (Python): The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.

  • fairseq

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

    Latest mention: Top 10 Developer Trends, Thu Oct 22 2020 | | 2020-10-22

    pytorch / fairseq

  • EasyOCR

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

  • nni

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

    Latest mention: Top 10 Developer Trends, Sun Oct 18 2020 | | 2020-10-18

    microsoft / nni

  • yolov5

    YOLOv5 in PyTorch > ONNX > CoreML > TFLite

    Latest mention: Yolov5x | | 2021-01-06

    Yes, it is trained on COCO. All the codes are from this repo

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

    Bringing Old Photo Back to Life (CVPR 2020 oral)

    Latest mention: Weekly Developer Roundup #23 - Sun Nov 22 2020 | | 2020-11-21

    microsoft/Bringing-Old-Photos-Back-to-Life (Python): Bringing Old Photo Back to Life (CVPR 2020 oral)

  • datasets

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

    Latest mention: [P] 611 text datasets in 467 languages in the new v1.2 release of HuggingFace datasets library | | 2021-01-05

    There will be 13 more bytthe end of this week, from Microsoft CodeXGlue, I had not the time to fix my PR earlier : .

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

    Latest mention: Apple’s New M1 Chip is a Machine Learning Beast | | 2020-12-25

    Yes. But that's missing the point. Almost everything in the space uses a framework other than Core ML. Therefore most people need good development support for PyTorch/Tensorflow etc., not Core ML. The fact that Apple has a tool to import/convert models is nice, but not relevant. Also, there is onnx as an exchange format between the frameworks, and tools like MMdnn and others.

  • stanza

    Official Stanford NLP Python Library for Many Human Languages

  • kornia

    Open Source Differentiable Computer Vision Library for PyTorch

  • espnet

    End-to-End Speech Processing Toolkit

    Latest mention: Downpour: DRM Free Audiobooks | | 2021-01-17

    I really like google text to speech and use it for my own custom audiobooks, I've tried google's microsoft's, IBM's, and a few other research ones. IBM's sounds slightly better but has a much more restricted free monthly tier, google's and microsoft's has 1 million free characters per month which goes pretty far.

    Like others are saying, it's slightly robotic but I've started to listen a ton by TTS and you definitely get used to it (you even start to hear inflection in it, which is cool). I use android smart audiobook app and you can control the sound levels, turning down the high pitch aspects also helps to make it easier to listen to for longer periods of time

    For HN folks, there are some pretty reasonable research projects, especially by nvidia (glownet) which you can run yourself. They sound relatively similar but the training voices are much more restricted and not as good. If anyone knows of a github/etc with a nicer diy TTS I'd be interested. The best I've seen which is customizable is but I had trouble getting it to work, then getting it to sound ok

    (For anyone else going DIY I'll warn you that the failure modes for TTS is some unerring frankly creepy sounds. Google's TTS fails very well, even for strange words, and when it gets very confused it spells it out. Some of the research ones go into haunting unrelated syllables, sometimes repeating for 10s of seconds

  • mindsdb

    Predictive AI layer for existing databases.

  • spotlight

    Deep recommender models using PyTorch.

  • hummingbird

    Hummingbird compiles trained ML models into tensor computation for faster inference.

    Latest mention: I learned about Microsoft's Hummingbird library today. 1000x performance?? | | 2020-09-23

    I took their sample code from Github and tweaked it to spit out times for each model's prediction, as well as increase the number of rows to 5 million. I used Google's Colab and selected GPU for my hardware accelerator. This gives an option to run code on GPU, not that all computations will happen on the GPU.

  • coremltools

    Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.

    Latest mention: Apple’s New M1 Chip is a Machine Learning Beast | | 2020-12-25

    And did you even know that Apple provides a CoreML conversion toolkit for converting models from PyTorch, TF, SKLearn and Keras to CoreML? You can comfortably train in your preferred tool and then convert the models to CoreML. ( It currently supports a lot of conversions

  • jina

    An easier way to build neural search in the cloud

    Latest mention: Thank you Hacktoberfest! | | 2020-11-09
  • PyTorch-NLP

    Basic Utilities for PyTorch Natural Language Processing (NLP)

  • pytorch-optimizer

    torch-optimizer -- collection of optimizers for Pytorch

    Latest mention: [R] AdasOptimizer Update: Cifar-100+MobileNetV2 Adas generalizes with Adas 15% better and 9x faster than Adam | | 2021-01-15
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).


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

Project Stars
1 transformers 39,321
2 Real-Time-Voice-Cloning 22,118
3 pytorch-tutorial 19,371
4 pytorch-CycleGAN-and-pix2pix 14,037
5 mmdetection 13,335
6 pytorch-lightning 11,417
7 fairseq 10,980
8 EasyOCR 9,975
9 nni 8,684
10 yolov5 7,781
11 Bringing-Old-Photos-Back-to-Life 7,293
12 datasets 6,482
13 MMdnn 5,166
14 stanza 5,091
15 kornia 3,509
16 espnet 3,308
17 mindsdb 3,146
18 spotlight 2,386
19 hummingbird 2,150
20 coremltools 2,079
21 jina 1,852
22 PyTorch-NLP 1,849
23 pytorch-optimizer 1,623