Python Pytorch

Open-source Python projects categorized as Pytorch

Top 23 Python Pytorch Projects

  • transformers

    🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

    Project mention: BetterTransformer: PyTorch-native free-lunch speedups for Transformer-based models | | 2022-11-22

    In order to support BetterTransformer with the canonical Transformer models from Transformers library, an integration was done with the open-source library Optimum as a one-liner:

  • Real-Time-Voice-Cloning

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

    Project mention: Speech to text | | 2022-09-29
  • Scout APM

    Truly a developer’s best friend. Scout APM is great for developers who want to find and fix performance issues in their applications. With Scout, we'll take care of the bugs so you can focus on building great things 🚀.

  • yolov5

    YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

    Project mention: YOLOv5 SOTA Realtime Instance Segmentation | | 2022-11-22
  • pytorch-tutorial

    PyTorch Tutorial for Deep Learning Researchers

    Project mention: PyTorch - What does contiguous() do? | | 2022-03-29

    I was going through this example of a LSTM language model on github (link).What it does in general is pretty clear to me. But I'm still struggling to understand what calling contiguous() does, which occurs several times in the code.

  • MockingBird

    🚀AI拟声: 5秒内克隆您的声音并生成任意语音内容 Clone a voice in 5 seconds to generate arbitrary speech in real-time

    Project mention: TIL cyber criminals with the help of A.I voice cloning software, used a deepfaked voice of a company executive to fool a Emirati bank manager to transfer 35 million dollars into their personal accounts. The bank manager had recognized the executive's voice from having worked with him before. | | 2022-09-12

    Actually, there are already open source implementations available, for example, the MockingBird project on GitHub. It supports English and Mandarin Chinese. For those with enough computation power and willingness to try, you can even make your own voice dataset and train the model to generate ‘your’ sound, simply following the project docs.


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

    Project mention: MM7 Characters ran through a face restoration algorithm | | 2022-11-07

    Link: To get these results I blurred the original images a bit before running through the algorithm, otherwise they look too 3D

  • mmdetection

    OpenMMLab Detection Toolbox and Benchmark

    Project mention: MMDeploy: Deploy All the Algorithms of OpenMMLab | | 2022-11-21

    MMDetection: OpenMMLab detection toolbox and benchmark.

  • Sonar

    Write Clean Python Code. Always.. Sonar helps you commit clean code every time. With over 225 unique rules to find Python bugs, code smells & vulnerabilities, Sonar finds the issues while you focus on the work.

  • pytorch-image-models

    PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

    Project mention: [D] Timm's implementation of stochastic depth | | 2022-11-11

    In Timm's implementation of stochastic depth ( the tensor is scaled by the probability of keeping the actual block. I didn't understand why he does so specially that this is not mentioned in the paper.

  • lightning

    Build and train PyTorch models and connect them to the ML lifecycle using Lightning App templates, without handling DIY infrastructure, cost management, scaling, and other headaches.

    Project mention: We just release a complete open-source solution for accelerating Stable Diffusion pretraining and fine-tuning! | | 2022-11-11

    Our codebase for the diffusion models builds heavily on OpenAI's ADM codebase , lucidrains, Stable Diffusion, Lightning and Hugging Face. Thanks for open-sourcing!

  • fairseq

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

    Project mention: [R] Speech-to-speech translation for a real-world unwritten language | | 2022-10-22


  • pytorch-CycleGAN-and-pix2pix

    Image-to-Image Translation in PyTorch

    Project mention: IA art | | 2022-09-26
  • Real-ESRGAN

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

    Project mention: Anyone have the full size image of this. It is from an LG TV. Possibly animated. Thought this is the best place to ask ? | | 2022-11-25

    Yes sir, spent ages trying to find the original with no luck, so I ran Real-ESRGAN on my PC and upscaled it to 4K that way.

  • EasyOCR

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

    Project mention: [P]Modern open-source OCR capabilities and which model to choose | | 2022-11-18

    I've used EasyOCR for number recognition tasks. Works fairly well.

  • pytorch_geometric

    Graph Neural Network Library for PyTorch

    Project mention: RL Agent Library to use graph in spaces | | 2022-10-22

    I don't know if any RL library includes an already implemented agent that can process graphs. However there are a number of deep learning frameworks that can help with the implementation of graph neural networks, especially Graph Nets (based on Tensorflow) and PyTorch Geometric. You might need to modify an existing RL agent to make use of one of these frameworks. If you are not familiar with GNNs you can look up these surveys. This article may also be of interest to you: it tackles graph-based environments, and the paper's code is available (it has a custom implementation of A2C and uses PyTorch Geometric -- btw it doesn't use Gym's space.graph since this feature is very recent in Gym).

  • d2l-en

    Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 400 universities from 60 countries including Stanford, MIT, Harvard, and Cambridge.

    Project mention: How to pre-train BERT on different objective tasks using HuggingFace | | 2022-04-10

    There might is bert library for pre-train bert model in huggingface, But I suggestion that you train bert model in native pytorch to understand detail, Limu's course is recommended for you

  • datasets

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

    Project mention: FauxPilot – an open-source GitHub Copilot server | | 2022-08-02

    And then pass that my_code.json as the dataset name.


  • insightface

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

    Project mention: hey guys which is the best tool for making facial recognition using single image in deep learning | | 2022-10-11

    If you're looking for models / planning to write some code yourself: Have a look at ArcFace for recognition and RetinaFace for face detection by Insightface. DeepFace repository might also be worth a try.

  • horovod

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

    Project mention: Anyone know of any papers or models for segmenting satellite images of a city into things like roads, buildings, parks, etc? | | 2022-04-25

    Training is not the same as inference (doing the segmentation), so that scale is probably off by a lot. One or two orders of magnitude just depending on the specifics of what hardware you're running on, and your training and eval dataset would be several orders of magnitude smaller. FAANGs would parallelize that training as well (don't remember if UNet is inherently parallelizable for training) via their internal equivalent of Horovod, so they'll do a GPU-month worth of training in less than a day.

  • nni

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

  • flair

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

    Project mention: Flair: A simple framework for state-of-the-art Natural Language Processing | | 2022-04-11
  • best-of-ml-python

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

    Project mention: Best-Of Machine Learning with Python | | 2022-04-28
  • MindsDB

    In-Database Machine Learning

    Project mention: Tutorial to Predict Amsterdam Housing Prices Using MindsDB and MongoDB | | 2022-10-31

    If you enjoyed following along to this tutorial, make sure to Sign Up for a free MindsDB Cloud account and continue exploring! You can also check them out on GitHub.

  • allennlp

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

    Project mention: How to solve ConfigurationError using HuggingFace Token Classifier | | 2022-10-08

    No clue. So what I did was google the error. Here's what I found:

  • Zigi

    Delete the most useless function ever: context switching.. Zigi monitors Jira and GitHub updates, pings you when PRs need approval and lets you take fast actions - all directly from Slack! Plus it reduces cycle time by up to 75%.

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 2022-11-25.

Python Pytorch related posts


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

Project Stars
1 transformers 74,719
2 Real-Time-Voice-Cloning 37,833
3 yolov5 33,073
4 pytorch-tutorial 25,251
5 MockingBird 25,060
6 GFPGAN 24,795
7 mmdetection 22,140
8 pytorch-image-models 22,094
9 lightning 20,706
10 fairseq 19,942
11 pytorch-CycleGAN-and-pix2pix 18,779
12 Real-ESRGAN 16,592
13 EasyOCR 16,385
14 pytorch_geometric 16,152
15 d2l-en 15,632
16 datasets 14,783
17 insightface 12,964
18 horovod 12,854
19 nni 12,254
20 flair 12,235
21 best-of-ml-python 11,952
22 MindsDB 11,326
23 allennlp 11,300
Build time-series-based applications quickly and at scale.
InfluxDB is the Time Series Data Platform where developers build real-time applications for analytics, IoT and cloud-native services in less time with less code.