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,624 40,746 4,777 (49 issues need help) 1,679 Updated Aug 4, 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 354 2,968 54 (20 issues need help) 23 Updated Aug 4, 2020

text

Data loaders and abstractions for text and NLP

nlp deep-learning pytorch dataset data-loader

Python BSD-3-Clause 549 2,400 175 (14 issues need help) 35 Updated Aug 4, 2020

vision

Datasets, Transforms and Models specific to Computer Vision

machine-learning computer-vision

Jupyter Notebook BSD-3-Clause 3,561 6,954 269 (71 issues need help) 125 Updated Aug 4, 2020

audio

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

audio python mp3 speech wav io

Python BSD-2-Clause 221 1,038 88 (13 issues need help) 35 Updated Aug 4, 2020

FBGEMM

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

C++ 151 493 7 5 Updated Aug 4, 2020

tensorpipe

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

C++ 10 54 13 9 Updated Aug 4, 2020

cppdocs

PyTorch C++ API Documentation

HTML 18 93 4 3 Updated Aug 4, 2020

tutorials

PyTorch tutorials.

Jupyter Notebook BSD-3-Clause 2,155 4,100 186 154 Updated Aug 4, 2020

glow

Compiler for Neural Network hardware accelerators

C++ Apache-2.0 455 2,211 234 (1 issue needs help) 26 Updated Aug 4, 2020

xla

Enabling PyTorch on Google TPU

C++ 129 950 49 4 Updated Aug 4, 2020

serve

Model Serving on PyTorch

Java Apache-2.0 127 1,099 79 (1 issue needs help) 24 Updated Aug 4, 2020

botorch

Bayesian optimization in PyTorch

Python MIT 146 1,660 24 6 Updated Aug 4, 2020

pytorch.github.io

The website for PyTorch

Jupyter Notebook BSD-3-Clause 121 101 33 21 Updated Aug 4, 2020

fairseq

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

python pytorch artificial-intelligence

Python MIT 2,148 8,589 319 (72 issues need help) 50 Updated Aug 4, 2020

builder

Continuous builder and binary build scripts for pytorch

Shell BSD-2-Clause 86 124 10 11 Updated Aug 4, 2020

hub

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

Python 106 953 11 1 Updated Aug 3, 2020

examples

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

Python BSD-3-Clause 6,441 13,785 202 62 Updated Aug 3, 2020

ios-demo-app

PyTorch iOS examples

Swift 40 112 8 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

rfcs

PyTorch RFCs (experimental)

2 11 1 5 Updated Jul 31, 2020

captum

Model interpretability and understanding for PyTorch

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

Python BSD-3-Clause 146 1,631 21 (1 issue needs help) 4 Updated Jul 30, 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 413 5 7 Updated Jul 30, 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

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 1 Updated Jul 25, 2020

nestedtensor

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

Python 7 75 39 1 Updated Jul 23, 2020

elastic

PyTorch elastic training

Python BSD-3-Clause 43 532 6 2 Updated Jul 22, 2020

dr-ci

Diagnose and remediate CI jobs

Haskell MIT 0 2 2 0 Updated Jul 21, 2020