pytorch

Repositories

• glow

  Compiler for Neural Network hardware accelerators
  C++ Apache-2.0 422 2,135 209 (1 issue needs help) 15 Updated May 20, 2020

• pytorch

  Tensors and Dynamic neural networks in Python with strong GPU acceleration

  python machine-learning deep-learning neural-network gpu numpy autograd
  C++ 9,959 38,788 4,371 (18 issues need help) 1,452 Updated May 20, 2020

• pytorch.github.io

  The website for PyTorch
  Jupyter Notebook BSD-3-Clause 108 89 23 13 Updated May 20, 2020

• text

  Data loaders and abstractions for text and NLP

  nlp deep-learning pytorch dataset data-loader
  Python BSD-3-Clause 526 2,289 157 (13 issues need help) 31 Updated May 20, 2020

• serve

  Model Serving on PyTorch
  Java Apache-2.0 76 811 93 (2 issues need help) 23 Updated May 20, 2020

• cppdocs

  PyTorch C++ API Documentation
  HTML 18 85 4 3 Updated May 20, 2020

• vision

  Datasets, Transforms and Models specific to Computer Vision

  machine-learning computer-vision
  Jupyter Notebook BSD-3-Clause 3,228 6,271 235 (64 issues need help) 102 Updated May 20, 2020

• xla

  Enabling PyTorch on Google TPU
  C++ 114 833 50 (1 issue needs help) 5 Updated May 20, 2020

• ignite

  High-level library to help with training neural networks in PyTorch

  python machine-learning deep-learning neural-network pytorch
  Python BSD-3-Clause 318 2,813 49 (20 issues need help) 21 Updated May 20, 2020

• hub

  Submission to https://pytorch.org/hub/
  Python 95 928 11 0 Updated May 19, 2020

• tensorpipe

  A tensor-aware point-to-point communication primitive for machine learning
  C++ 7 31 15 6 Updated May 19, 2020

• audio

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

  audio python mp3 speech wav io
  Python BSD-2-Clause 192 937 65 (6 issues need help) 25 Updated May 19, 2020

• ios-demo-app

  PyTorch iOS examples
  Swift 33 92 1 0 Updated May 19, 2020

• botorch

  Bayesian optimization in PyTorch
  Python MIT 133 1,594 18 5 Updated May 19, 2020

• tutorials

  PyTorch tutorials.
  Jupyter Notebook BSD-3-Clause 1,972 3,810 164 122 Updated May 19, 2020

• fairseq

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

  python pytorch artificial-intelligence
  Python MIT 1,941 7,903 212 (64 issues need help) 41 Updated May 19, 2020

• builder

  Continuous builder and binary build scripts for pytorch
  Shell BSD-2-Clause 78 110 9 9 Updated May 19, 2020

• FBGEMM

  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  C++ 143 463 8 4 Updated May 18, 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 114 373 5 6 Updated May 17, 2020

• nestedtensor

  [EXPERIMENTAL] Tools for the concurrent manipulation of variably sized Tensors.
  Python 7 52 39 2 Updated May 16, 2020

• translate

  Translate - a PyTorch Language Library

  machine-learning pytorch artificial-intelligence onnx
  Python BSD-3-Clause 139 622 15 48 Updated May 15, 2020

• captum

  Model interpretability and understanding for PyTorch

  interpretability feature-importance interpretable-ai interpretable-ml feature-attribution
  Python BSD-3-Clause 121 1,419 19 (1 issue needs help) 3 Updated May 15, 2020

• elastic

  PyTorch elastic training
  Python BSD-3-Clause 37 435 4 2 Updated May 12, 2020

• sccache

  Forked from mozilla/sccache

  sccache is ccache with cloud storage
  Rust Apache-2.0 178 1 0 0 Updated May 6, 2020

• pytorch-probot

  PyTorch GitHub bot written in probot
  TypeScript ISC 2 8 1 2 Updated May 6, 2020

• pytorch_sphinx_theme

  PyTorch Sphinx Theme
  CSS MIT 30 30 2 8 Updated May 4, 2020

• examples

  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  Python BSD-3-Clause 6,095 12,947 194 61 Updated May 4, 2020

• ossci-job-dsl

  Jenkins job definitions for OSSCI
  Groovy Apache-2.0 28 6 0 2 Updated Apr 27, 2020

• rfcs

  PyTorch RFCs (experimental)
  2 8 1 5 Updated Apr 27, 2020

• extension-script

  Example repository for custom C++/CUDA operators for TorchScript

  cpp cuda pytorch
  Python 8 63 4 2 Updated Apr 17, 2020