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,686 40,886 4,803 (49 issues need help) 1,696 Updated Aug 10, 2020

vision

Datasets, Transforms and Models specific to Computer Vision

machine-learning computer-vision

Jupyter Notebook BSD-3-Clause 3,586 7,000 278 (72 issues need help) 122 Updated Aug 10, 2020

serve

Model Serving on PyTorch

Java Apache-2.0 131 1,120 74 (1 issue needs help) 23 Updated Aug 10, 2020

audio

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

audio python mp3 speech wav io

Python BSD-2-Clause 225 1,044 92 (13 issues need help) 32 Updated Aug 10, 2020

text

Data loaders and abstractions for text and NLP

nlp deep-learning pytorch dataset data-loader

Python BSD-3-Clause 549 2,407 175 (14 issues need help) 34 Updated Aug 10, 2020

fairseq

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

python pytorch artificial-intelligence

Python MIT 2,164 8,633 337 (72 issues need help) 46 Updated Aug 10, 2020

rfcs

PyTorch RFCs (experimental)

2 11 1 5 Updated Aug 10, 2020

glow

Compiler for Neural Network hardware accelerators

C++ Apache-2.0 457 2,219 232 (1 issue needs help) 16 Updated Aug 10, 2020

captum

Model interpretability and understanding for PyTorch

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

Python BSD-3-Clause 152 1,640 21 (1 issue needs help) 3 Updated Aug 10, 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 357 2,971 53 (18 issues need help) 23 Updated Aug 10, 2020

pytorch.github.io

The website for PyTorch

Jupyter Notebook BSD-3-Clause 122 101 34 23 Updated Aug 10, 2020

tutorials

PyTorch tutorials.

Jupyter Notebook BSD-3-Clause 2,163 4,124 191 158 Updated Aug 10, 2020

xla

Enabling PyTorch on Google TPU

C++ 132 959 46 6 Updated Aug 9, 2020

botorch

Bayesian optimization in PyTorch

Python MIT 147 1,667 26 6 Updated Aug 9, 2020

examples

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

Python BSD-3-Clause 6,459 13,838 204 63 Updated Aug 8, 2020

tensorpipe

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

C++ 10 56 14 4 Updated Aug 8, 2020

FBGEMM

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

C++ 152 494 7 6 Updated Aug 8, 2020

builder

Continuous builder and binary build scripts for pytorch

Shell BSD-2-Clause 87 126 10 14 Updated Aug 7, 2020

hub

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

Python 106 953 11 3 Updated Aug 7, 2020

nestedtensor

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

Python 7 76 39 2 Updated Aug 7, 2020

translate

Translate - a PyTorch Language Library

machine-learning pytorch artificial-intelligence onnx

Python BSD-3-Clause 154 640 15 48 Updated Aug 4, 2020

cppdocs

PyTorch C++ API Documentation

HTML 18 93 4 3 Updated Aug 4, 2020

ios-demo-app

PyTorch iOS examples

Swift 40 113 7 0 Updated Aug 3, 2020

ossci-job-dsl

Jenkins job definitions for OSSCI

Groovy Apache-2.0 30 7 0 2 Updated Aug 3, 2020

pytorch_sphinx_theme

PyTorch Sphinx Theme

CSS MIT 36 33 2 11 Updated Jul 30, 2020

cpuinfo

CPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)

cpu cpuid cpu-cache cpu-model instruction-set cpu-topology

Objective-C BSD-2-Clause 128 415 5 6 Updated Jul 30, 2020

workshops

This is a repository for all workshop related materials.

8 11 0 2 Updated Jul 28, 2020

pytorch-integration-testing

Testing downstream libraries using pytorch release candidates

Dockerfile MIT 1 2 0 0 Updated Jul 25, 2020

elastic

PyTorch elastic training

Python BSD-3-Clause 44 536 5 2 Updated Jul 22, 2020

dr-ci

Diagnose and remediate CI jobs

Haskell MIT 0 2 2 0 Updated Jul 21, 2020