81 Open source projects that are alternatives of or similar to PyTorch

Code for CheXpert Challenge 2019 Top 1 && Top 2 solution

Pytorch Implementation of the paper - "Tidying Deep Saliency Prediction Architectures"

[TPAMI Under Submission] Generative Multi-Label Zero-Shot Learning

A clean and lucid implementation of cycleGAN using PyTorch

Standard federated learning implementations in FedLab and FL benchmarks.

Implementation of Sparse Shift Layer and Active Shift Layer (3D, 4D, 5D tensors) for PyTorch(CPU,GPU)

A PyTorch Implementation of Convolutional Conditional Neural Process.

Its a social networking chat-bot trained on Reddit dataset . It supports open bounded queries developed on the concept of Neural Machine Translation. Beware of its being sarcastic just like its creator 😝 BDW it uses Pytorch framework and Python3.

PyTorch implementation of ESRGAN (ECCVW 2018) for compressed image subjective quality enhancement.

Reproduce some methods in semi-supervised papers.

PyTorch Implementation for Neural Graph Collaborative Filtering

This is a PyTorch implementation of the paper: "Processing Megapixel Images with Deep Attention-Sampling Models".

Code and dataset for the IJCAI 2021 paper "Document-level Relation Extraction as Semantic Segmentation".

The implementation of 3D-UNet using PyTorch

PyTorch implementation of EfficientNetV2 family

ResUNet++, CRF, and TTA for segmentation of medical images (IEEE JBIHI)

Awesome Pytorch Tutorials

[CVPR2020] Adversarial Latent Autoencoders

[ECCV 2020] Official Pytorch implementation for "Latent Embedding Feedback and Discriminative Features for Zero-Shot Classification". SOTA results for ZSL and GZSL

Pytorch implementation of Paragraph-level Neural Question Generation with Maxout Pointer and Gated Self-attention Networks

If you've always wanted to learn about deep-learning but don't know where to start, then you might have stumbled upon the right place!

PyTorch implementation of 'An Unsupervised Neural Attention Model for Aspect Extraction' by He et al. ACL2017'

Implementation of DeepMind's Deep Generative Model of Radar (DGMR) https://arxiv.org/abs/2104.00954

PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results

This repo is official PyTorch implementation of MobileHumanPose: Toward real-time 3D human pose estimation in mobile devices(CVPRW 2021).

[ WSDM '19 ] Sequential Variational Autoencoders for Collaborative Filtering

Implementation of MelNet in PyTorch to generate high-fidelity audio samples

Highly efficient PyTorch version of the Semi-hard Triplet loss ⚡️

Pytorch implementation of Paper by Google Research - Representation Learning for Information Extraction from Form-like Documents.

[AAAI 2020] Code release for "Domain Conditioned Adaptation Network" https://arxiv.org/abs/2005.06717

A PyTorch Implementation of "Optimization of Molecules via Deep Reinforcement Learning".

Source code for our "TitleStylist" paper at ACL 2020

🔥 OpenPose api wrapper in PyTorch.

A tool for precisely placing 3D landmarks on 3D facial scans based on the paper "Multi-view Consensus CNN for 3D Facial Landmark Placement"

Neuro-Symbolic Visual Question Answering on Sort-of-CLEVR using PyTorch

A repo containing code covering various aspects of deep learning on Pytorch. Great for beginners and intermediate in the field

[MICCAI2022] This is an official PyTorch implementation for A Robust Volumetric Transformer for Accurate 3D Tumor Segmentation

Code for the ACL 19 paper "Neural Keyphrase Generation via Reinforcement Learning with Adaptive Rewards"

A Collection of Variational Autoencoders (VAE) in PyTorch.

ViNet Pushing the limits of Visual Modality for Audio Visual Saliency Prediction

Tutorials on implementing a few sequence-to-sequence (seq2seq) models with PyTorch and TorchText.

Pytorch Implementation of Depth Map Prediction from a Single Image using a Multi-Scale Deep Network

Easy-to-follow Pytorch tutorial Notebook for Multi-Agent-Diverse-Generative-Adversarial-Networks

PyTorch implementation of GANs (Generative Adversarial Networks). DCGAN, Pix2Pix, CycleGAN, SRGAN

[UNMAINTAINED] A starter pack for creating a lightweight responsive web app for Fast.AI PyTorch models.

nnDetection is a self-configuring framework for 3D (volumetric) medical object detection which can be applied to new data sets without manual intervention. It includes guides for 12 data sets that were used to develop and evaluate the performance of the proposed method.

The project is the official implementation of our CVPR 2021 paper, "Restoring Extremely Dark Images in Real Time"

Hyperparameter-Free Out-of-Distribution Detection Using Softmax of Scaled Cosine Similarity

Learning low-shot object classification with explicit shape bias learned from point clouds

Pytorch implementation of the Loc2Vec with some modifications for speed

"Self-training with Noisy Student improves ImageNet classification" pytorch implementation

🍀 Pytorch Implementation of RandLA-Net (https://arxiv.org/abs/1911.11236)

Metrics for model evaluation in pytorch

Topic clustering library built on Transformer embeddings and cosine similarity metrics.Compatible with all BERT base transformers from huggingface.

The aim of this repository is to show a baseline model for text classification by implementing a LSTM-based model coded in PyTorch. In order to provide a better understanding of the model, it will be used a Tweets dataset provided by Kaggle.

VNet for 3d volume segmentation

An unofficial toy implementation for NVAE 《A Deep Hierarchical Variational Autoencoder》

Implementation of Resnet-50 with and without CBAM in PyTorch v1.8. Implementation tested on Intel Image Classification dataset from https://www.kaggle.com/puneet6060/intel-image-classification.

Implement of Dynamic Model Pruning with Feedback with pytorch

Online Deep Learning: Learning Deep Neural Networks on the Fly / Non-linear Contextual Bandit Algorithm (ONN_THS)