Top 3972 pytorch open source projects

Toolbox for converting analog to spiking neural networks (ANN to SNN), and running them in a spiking neuron simulator.

Pytorch implementation of Tacotron

PyTorch Framework for Developing Memory Efficient Deep Invertible Networks

Code for Hands On Intelligent Agents with OpenAI Gym book to get started and learn to build deep reinforcement learning agents using PyTorch

A PyTorch port of Google TensorFlow.js PoseNet (Real-time Human Pose Estimation)

Sentiment analysis neural network trained by fine-tuning BERT, ALBERT, or DistilBERT on the Stanford Sentiment Treebank.

A OpenMMLAB toolbox for human pose estimation, skeleton-based action recognition, and action synthesis.

Unofficial PyTorch Implementation of SUM-GAN from "Unsupervised Video Summarization with Adversarial LSTM Networks" (CVPR 2017)

Official PyTorch Implementation of Proxy Anchor Loss for Deep Metric Learning, CVPR 2020

3D Graph Neural Networks for RGBD Semantic Segmentation

PyTorch implementation of VQ-VAE + WaveNet by [Chorowski et al., 2019] and VQ-VAE on speech signals by [van den Oord et al., 2017]

GraphGallery is a gallery for benchmarking Graph Neural Networks (GNNs) and Graph Adversarial Learning with TensorFlow 2.x and PyTorch backend.

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

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

An easy to use PyTorch to TensorRT converter

[ICCV 2019] Monocular depth estimation from a single image

一个可以自己进行训练的中文聊天机器人， 根据自己的语料训练出自己想要的聊天机器人，可以用于智能客服、在线问答、智能聊天等场景。目前包含seq2seq、seqGAN版本、tf2.0版本、pytorch版本。

A hyperparameter optimization and data collection toolbox for convenient and fast prototyping of machine-learning models.

Rank Consistent Ordinal Regression for Neural Networks with Application to Age Estimation

Attention over nodes in Graph Neural Networks using PyTorch (NeurIPS 2019)

A pytorch implementation of the ACL2019 paper "Simple and Effective Text Matching with Richer Alignment Features".

科大讯飞2020事件抽取挑战赛第一名解决方案&完整事件抽取系统

CRSLab is an open-source toolkit for building Conversational Recommender System (CRS).

Global-Locally Self-Attentive Dialogue State Tracker

CGNet: A Light-weight Context Guided Network for Semantic Segmentation [IEEE Transactions on Image Processing 2020]

A multi-purpose Video Labeling GUI in Python with integrated SOTA detector and tracker

Neural abstractive summarization (seq2seq + copy (or pointer network) + coverage) in pytorch on CNN/Daily Mail

Hierarchical Attention Networks | a PyTorch Tutorial to Text Classification

A library for ready-made reinforcement learning agents and reusable components for neat prototyping

Protein Graph Library

A cleaner way to build neural networks for PyTorch.

Source code of CVPR 2020 paper, "HOPE-Net: A Graph-based Model for Hand-Object Pose Estimation"

Seq2Seq, Bert, Transformer, WaveNet for time series prediction.

[CVPR 2020] Estimation of the visible and hidden traversable space from a single color image

Dilated RNNs in pytorch

A PyTorch Implementation of "Neural Arithmetic Logic Units"

Distributed DAG (Directed acyclic graph) framework for machine learning with UI

a sklearn wrapper for Google's BERT model

Activity Recognition Algorithms for the Charades Dataset

top 1% solution to toxic comment classification challenge on Kaggle.

Fully reproduce the paper of StarGAN-VC. Stable training and Better audio quality .

Naszilla is a Python library for neural architecture search (NAS)

Pytorch implementation of "One-Sided Unsupervised Domain Mapping" NIPS 2017

PyTorch implementation of 1712.06087 "Zero-Shot" Super-Resolution using Deep Internal Learning

The pytorch code of the paper "Intrinsic Image Popularity Assessment"

Just draw a bounding box and you can remove the object you want to remove.

Transformer-based image captioning extension for pytorch/fairseq

A set of tutorials for torchtext

This repository contains the source code and data for reproducing results of Deep Continuous Clustering paper

E(2)-Equivariant CNNs Library for Pytorch

Fast PyTorch based DSP for audio and 1D signals

This is code of book "Learn Deep Learning with PyTorch"

Just another DL library

My completed implementation solutions for CS224N 2019

Vocal Remover using Deep Neural Networks

PyTorch-based modular, configuration-driven framework for knowledge distillation. 🏆18 methods including SOTA are implemented so far. 🎁 Trained models, training logs and configurations are available for ensuring the reproducibiliy.

Machine learning metrics for distributed, scalable PyTorch applications.

Image Super-Resolution Using SRCNN, DRRN, SRGAN, CGAN in Pytorch

Tutorial about 3D convolutional network