bharathgs / Awesome-pytorch-list

Awesome-Pytorch-list ======================== Contents • Pytorch & related libraries • NLP & Speech Processing • Computer Vision • Probabilistic/Generative Libraries • Other libraries • Tutorials, books & examples • Paper implementations • Talks & Conferences • Pytorch elsewhere Pytorch & related libraries • pytorch: Tensors and Dynamic neural networks in Python with strong GPU acceleration. • Captum: Model interpretability and understanding for PyTorch. NLP & Speech Processing: • pytorch text: Torch text related contents. • pytorch-seq2seq: A framework for sequence-to-sequence (seq2seq) models implemented in PyTorch. • anuvada: Interpretable Models for NLP using PyTorch. • audio: simple audio I/O for pytorch. • loop: A method to generate speech across multiple speakers • fairseq-py: Facebook AI Research Sequence-to-Sequence Toolkit written in Python. • speech: PyTorch ASR Implementation. • OpenNMT-py: Open-Source Neural Machine Translation in PyTorch http://opennmt.net • neuralcoref: State-of-the-art coreference resolution based on neural nets and spaCy huggingface.co/coref • sentiment-discovery: Unsupervised Language Modeling at scale for robust sentiment classification. • MUSE: A library for Multilingual Unsupervised or Supervised word Embeddings • nmtpytorch: Neural Machine Translation Framework in PyTorch. • pytorch-wavenet: An implementation of WaveNet with fast generation • Tacotron-pytorch: Tacotron: Towards End-to-End Speech Synthesis. • AllenNLP: An open-source NLP research library, built on PyTorch. • PyTorch-NLP: Text utilities and datasets for PyTorch pytorchnlp.readthedocs.io • quick-nlp: Pytorch NLP library based on FastAI. • TTS: Deep learning for Text2Speech • LASER: Language-Agnostic SEntence Representations • pyannote-audio: Neural building blocks for speaker diarization: speech activity detection, speaker change detection, speaker embedding • gensen: Learning General Purpose Distributed Sentence Representations via Large Scale Multi-task Learning. • translate: Translate - a PyTorch Language Library. • espnet: End-to-End Speech Processing Toolkit espnet.github.io/espnet • pythia: A software suite for Visual Question Answering • UnsupervisedMT: Phrase-Based & Neural Unsupervised Machine Translation. • jiant: The jiant sentence representation learning toolkit. • BERT-PyTorch: Pytorch implementation of Google AI's 2018 BERT, with simple annotation • InferSent: Sentence embeddings (InferSent) and training code for NLI. • uis-rnn:This is the library for the Unbounded Interleaved-State Recurrent Neural Network (UIS-RNN) algorithm, corresponding to the paper Fully Supervised Speaker Diarization. arxiv.org/abs/1810.04719 • flair: A very simple framework for state-of-the-art Natural Language Processing (NLP) • pytext: A natural language modeling framework based on PyTorch fb.me/pytextdocs • voicefilter: Unofficial PyTorch implementation of Google AI's VoiceFilter system http://swpark.me/voicefilter • BERT-NER: Pytorch-Named-Entity-Recognition-with-BERT. • transfer-nlp: NLP library designed for flexible research and development • texar-pytorch: Toolkit for Machine Learning and Text Generation, in PyTorch texar.io • pytorch-kaldi: pytorch-kaldi is a project for developing state-of-the-art DNN/RNN hybrid speech recognition systems. The DNN part is managed by pytorch, while feature extraction, label computation, and decoding are performed with the kaldi toolkit. • NeMo: Neural Modules: a toolkit for conversational AI nvidia.github.io/NeMo • pytorch-struct: A library of vectorized implementations of core structured prediction algorithms (HMM, Dep Trees, CKY, ..,) • espresso: Espresso: A Fast End-to-End Neural Speech Recognition Toolkit • transformers: huggingface Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch. huggingface.co/transformers • reformer-pytorch: Reformer, the efficient Transformer, in Pytorch • torch-metrics: Metrics for model evaluation in pytorch • speechbrain: SpeechBrain is an open-source and all-in-one speech toolkit based on PyTorch. • Backprop: Backprop makes it simple to use, finetune, and deploy state-of-the-art ML models. CV: • pytorch vision: Datasets, Transforms and Models specific to Computer Vision. • pt-styletransfer: Neural style transfer as a class in PyTorch. • OpenFacePytorch: PyTorch module to use OpenFace's nn4.small2.v1.t7 model • img_classification_pk_pytorch: Quickly comparing your image classification models with the state-of-the-art models (such as DenseNet, ResNet, ...) • SparseConvNet: Submanifold sparse convolutional networks. • Convolution_LSTM_pytorch: A multi-layer convolution LSTM module • face-alignment: :fire: 2D and 3D Face alignment library build using pytorch adrianbulat.com • pytorch-semantic-segmentation: PyTorch for Semantic Segmentation. • RoIAlign.pytorch: This is a PyTorch version of RoIAlign. This implementation is based on crop_and_resize and supports both forward and backward on CPU and GPU. • pytorch-cnn-finetune: Fine-tune pretrained Convolutional Neural Networks with PyTorch. • detectorch: Detectorch - detectron for PyTorch • Augmentor: Image augmentation library in Python for machine learning. http://augmentor.readthedocs.io • s2cnn: This library contains a PyTorch implementation of the SO(3) equivariant CNNs for spherical signals (e.g. omnidirectional cameras, signals on the globe) • TorchCV: A PyTorch-Based Framework for Deep Learning in Computer Vision. • maskrcnn-benchmark: Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch. • image-classification-mobile: Collection of classification models pretrained on the ImageNet-1K. • medicaltorch: A medical imaging framework for Pytorch http://medicaltorch.readthedocs.io • albumentations: Fast image augmentation library. • kornia: Differentiable computer vision library. • pytorch-text-recognition: Text recognition combo - CRAFT + CRNN. • facenet-pytorch: Pretrained Pytorch fac…