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,722 40,992 4,840 (49 issues need help) 1,718 Updated Aug 15, 2020

• glow

  Compiler for Neural Network hardware accelerators
  C++ Apache-2.0 461 2,221 233 (1 issue needs help) 15 Updated Aug 15, 2020

• captum

  Model interpretability and understanding for PyTorch

  interpretability feature-importance interpretable-ai interpretable-ml feature-attribution
  Python BSD-3-Clause 153 1,649 22 (1 issue needs help) 6 Updated Aug 15, 2020

• serve

  Model Serving on PyTorch
  Java Apache-2.0 134 1,134 76 (1 issue needs help) 23 Updated Aug 14, 2020

• fairseq

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

  python pytorch artificial-intelligence
  Python MIT 2,175 8,658 345 (73 issues need help) 48 Updated Aug 14, 2020

• xla

  Enabling PyTorch on Google TPU
  C++ 135 959 52 5 Updated Aug 14, 2020

• botorch

  Bayesian optimization in PyTorch
  Python MIT 148 1,668 25 6 Updated Aug 14, 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 361 2,977 55 (20 issues need help) 23 Updated Aug 14, 2020

• nestedtensor

  [EXPERIMENTAL] Tools for the concurrent manipulation of variably sized Tensors.
  Python 8 77 39 3 Updated Aug 14, 2020

• text

  Data loaders and abstractions for text and NLP

  nlp deep-learning pytorch dataset data-loader
  Python BSD-3-Clause 548 2,412 178 (14 issues need help) 37 Updated Aug 14, 2020

• vision

  Datasets, Transforms and Models specific to Computer Vision

  machine-learning computer-vision
  Jupyter Notebook BSD-3-Clause 3,605 7,035 283 (72 issues need help) 126 Updated Aug 14, 2020

• audio

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

  audio python mp3 speech wav io
  Python BSD-2-Clause 228 1,048 93 (13 issues need help) 36 Updated Aug 14, 2020

• tutorials

  PyTorch tutorials.
  Jupyter Notebook BSD-3-Clause 2,171 4,136 193 157 Updated Aug 14, 2020

• rfcs

  PyTorch RFCs (experimental)
  2 11 1 5 Updated Aug 14, 2020

• cppdocs

  PyTorch C++ API Documentation
  HTML 18 93 4 3 Updated Aug 14, 2020

• pytorch.github.io

  The website for PyTorch
  Jupyter Notebook BSD-3-Clause 121 101 34 21 Updated Aug 14, 2020

• FBGEMM

  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  C++ 153 494 7 8 Updated Aug 13, 2020

• builder

  Continuous builder and binary build scripts for pytorch
  Shell BSD-2-Clause 87 126 10 16 Updated Aug 14, 2020

• examples

  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  Python BSD-3-Clause 6,476 13,884 205 62 Updated Aug 12, 2020

• tensorpipe

  A tensor-aware point-to-point communication primitive for machine learning
  C++ 10 58 15 5 Updated Aug 13, 2020

• translate

  Translate - a PyTorch Language Library

  machine-learning pytorch artificial-intelligence onnx
  Python BSD-3-Clause 154 640 15 48 Updated Aug 12, 2020

• workshops

  This is a repository for all workshop related materials.
  8 12 0 2 Updated Aug 11, 2020

• hub

  Submission to https://pytorch.org/hub/
  Python 107 954 11 3 Updated Aug 11, 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 417 5 7 Updated Jul 30, 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 540 6 2 Updated Jul 22, 2020

• dr-ci

  Diagnose and remediate CI jobs
  Haskell MIT 0 2 2 0 Updated Jul 22, 2020