pytorch

Repositories

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

python machine-learning deep-learning neural-network gpu numpy autograd

C++ 10,584 40,595 4,732 (51 issues need help) 1,658 Updated Jul 30, 2020

serve

Model Serving on PyTorch

Java Apache-2.0 124 1,082 73 (1 issue needs help) 24 Updated Jul 30, 2020

vision

Datasets, Transforms and Models specific to Computer Vision

machine-learning computer-vision

Jupyter Notebook BSD-3-Clause 3,539 6,899 272 (69 issues need help) 132 Updated Jul 30, 2020

botorch

Bayesian optimization in PyTorch

Python MIT 143 1,655 23 4 Updated Jul 30, 2020

text

Data loaders and abstractions for text and NLP

nlp deep-learning pytorch dataset data-loader

Python BSD-3-Clause 550 2,391 177 (14 issues need help) 35 Updated Jul 30, 2020

glow

Compiler for Neural Network hardware accelerators

C++ Apache-2.0 456 2,206 229 (1 issue needs help) 21 Updated Jul 30, 2020

audio

Data manipulation and transformation for audio signal processing, powered by PyTorch

audio python mp3 speech wav io

Python BSD-2-Clause 219 1,030 88 (14 issues need help) 33 Updated Jul 30, 2020

FBGEMM

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

C++ 151 490 7 6 Updated Jul 29, 2020

tutorials

PyTorch tutorials.

Jupyter Notebook BSD-3-Clause 2,143 4,084 183 149 Updated Jul 29, 2020

xla

Enabling PyTorch on Google TPU

C++ 128 944 50 5 Updated Jul 29, 2020

tensorpipe

A tensor-aware point-to-point communication primitive for machine learning

C++ 10 52 14 5 Updated Jul 29, 2020

fairseq

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

python pytorch artificial-intelligence

Python MIT 2,134 8,563 305 (70 issues need help) 49 Updated Jul 29, 2020

cppdocs

PyTorch C++ API Documentation

HTML 18 93 4 3 Updated Jul 29, 2020

translate

Translate - a PyTorch Language Library

machine-learning pytorch artificial-intelligence onnx

Python BSD-3-Clause 153 640 15 48 Updated Jul 29, 2020

hub

Submission to https://pytorch.org/hub/

Python 103 950 11 0 Updated Jul 29, 2020

pytorch.github.io

The website for PyTorch

Jupyter Notebook BSD-3-Clause 119 101 33 19 Updated Jul 28, 2020

workshops

This is a repository for all workshop related materials.

6 11 0 2 Updated Jul 28, 2020

captum

Model interpretability and understanding for PyTorch

interpretability feature-importance interpretable-ai interpretable-ml feature-attribution

Python BSD-3-Clause 146 1,624 21 (1 issue needs help) 8 Updated Jul 28, 2020

ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

python machine-learning deep-learning neural-network pytorch

Python BSD-3-Clause 348 2,960 50 (18 issues need help) 23 Updated Jul 27, 2020

rfcs

PyTorch RFCs (experimental)

2 11 1 5 Updated Jul 25, 2020

builder

Continuous builder and binary build scripts for pytorch

Shell BSD-2-Clause 84 120 10 16 Updated Jul 25, 2020

pytorch-integration-testing

Testing downstream libraries using pytorch release candidates

Dockerfile MIT 1 2 0 1 Updated Jul 25, 2020

examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

Python BSD-3-Clause 6,423 13,741 202 61 Updated Jul 24, 2020

nestedtensor

[EXPERIMENTAL] Tools for the concurrent manipulation of variably sized Tensors.

Python 7 75 39 1 Updated Jul 24, 2020

elastic

PyTorch elastic training

Python BSD-3-Clause 42 531 6 2 Updated Jul 22, 2020

dr-ci

Diagnose and remediate CI jobs

Haskell MIT 0 2 2 0 Updated Jul 21, 2020

labeler-github-action

GitHub action for labeling issues and pull requests based on conditions

TypeScript MIT 1 1 0 4 Updated Jul 18, 2020

add-annotations-github-action

A GitHub action to run clang-tidy and annotate failures

JavaScript MIT 5 9 2 1 Updated Jul 18, 2020

pytorch-probot

PyTorch GitHub bot written in probot

TypeScript ISC 2 8 1 3 Updated Jul 18, 2020

pytorch_sphinx_theme

PyTorch Sphinx Theme

CSS MIT 35 32 2 10 Updated Jul 16, 2020