pytorch

Repositories

• tutorials

  PyTorch tutorials.
  Jupyter Notebook BSD-3-Clause 1,701 3,382 142 88 Updated Feb 22, 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,157 36,279 3,881 (16 issues need help) 1,199 Updated Feb 22, 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 274 2,574 30 (10 issues need help) 14 Updated Feb 22, 2020

• tensorpipe

  A tensor-aware point-to-point communication primitive for machine learning
  C++ 4 19 9 2 Updated Feb 22, 2020

• vision

  Datasets, Transforms and Models specific to Computer Vision

  machine-learning computer-vision
  Jupyter Notebook BSD-3-Clause 2,785 5,602 210 (48 issues need help) 97 Updated Feb 22, 2020

• audio

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

  audio python mp3 speech wav io
  Python BSD-2-Clause 168 815 54 (1 issue needs help) 15 Updated Feb 22, 2020

• cppdocs

  PyTorch C++ API Documentation
  HTML 16 75 3 3 Updated Feb 22, 2020

• pytorch.github.io

  The website for PyTorch
  Jupyter Notebook BSD-3-Clause 96 84 19 7 Updated Feb 22, 2020

• glow

  Compiler for Neural Network hardware accelerators
  C++ Apache-2.0 372 2,002 203 (1 issue needs help) 26 Updated Feb 22, 2020

• captum

  Model interpretability and understanding for PyTorch

  interpretability feature-importance interpretable-ai interpretable-ml feature-attribution
  Python BSD-3-Clause 90 1,091 12 (1 issue needs help) 6 Updated Feb 22, 2020

• xla

  Enabling PyTorch on Google TPU
  C++ 94 655 23 3 Updated Feb 22, 2020

• examples

  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  Python BSD-3-Clause 5,616 11,963 181 68 Updated Feb 21, 2020

• elastic

  PyTorch elastic training
  Python BSD-3-Clause 23 295 10 3 Updated Feb 21, 2020

• fairseq

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

  python pytorch artificial-intelligence
  Python MIT 1,685 6,995 123 (46 issues need help) 27 Updated Feb 21, 2020

• nestedtensor

  [EXPERIMENTAL] Tools for the concurrent manipulation of variably sized Tensors.
  Python 6 24 5 4 Updated Feb 21, 2020

• FBGEMM

  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  C++ 72 358 9 10 Updated Feb 21, 2020

• text

  Data loaders and abstractions for text and NLP

  nlp deep-learning pytorch dataset data-loader
  Python BSD-3-Clause 501 2,160 144 (11 issues need help) 23 Updated Feb 22, 2020

• botorch

  Bayesian optimization in PyTorch
  Python MIT 106 1,483 15 7 Updated Feb 21, 2020

• ossci-job-dsl

  Jenkins job definitions for OSSCI
  Groovy Apache-2.0 26 6 0 2 Updated Feb 20, 2020

• builder

  Continuous builder and binary build scripts for pytorch
  Shell BSD-2-Clause 70 102 7 8 Updated Feb 19, 2020

• rfcs

  PyTorch RFCs (experimental)
  1 5 0 3 Updated Feb 16, 2020

• translate

  Translate - a PyTorch Language Library

  machine-learning pytorch artificial-intelligence onnx
  Python BSD-3-Clause 129 584 16 49 Updated Feb 14, 2020

• contrib

  Implementations of ideas from recent papers
  Python 30 320 13 6 Updated Feb 4, 2020

• java-demo
  Jupyter Notebook 0 20 1 0 Updated Jan 22, 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 97 349 3 4 Updated Jan 22, 2020

• android-demo-app

  PyTorch android examples of usage in applications
  Java 47 115 12 4 Updated Jan 21, 2020

• hub

  Submission to https://pytorch.org/hub/
  Python 85 912 10 0 Updated Jan 18, 2020

• pytorch_sphinx_theme

  PyTorch Sphinx Theme
  CSS MIT 23 25 3 6 Updated Jan 16, 2020

• accimage

  high performance image loading and augmenting routines mimicking PIL.Image interface
  C BSD-2-Clause 26 220 8 1 Updated Jan 16, 2020

• tnt

  Simple tools for logging and visualizing, loading and training

  python machine-learning reinforcement-learning deep-learning neural-network pytorch visdom
  Python BSD-3-Clause 152 1,113 21 (1 issue needs help) 6 Updated Jan 15, 2020