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++ 12,923 48,173 5,000+ (22 issues need help) 2,683 Updated May 14, 2021

• serve

  Model Serving on PyTorch
  Java Apache-2.0 291 1,774 91 (3 issues need help) 28 Updated May 14, 2021

• text

  Data loaders and abstractions for text and NLP

  nlp deep-learning pytorch dataset data-loader
  Python BSD-3-Clause 633 2,754 229 (15 issues need help) 74 Updated May 14, 2021

• kineto

  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  HTML 27 95 29 8 Updated May 14, 2021

• glow

  Compiler for Neural Network hardware accelerators
  C++ Apache-2.0 550 2,467 255 (2 issues need help) 100 Updated May 14, 2021

• pytorch.github.io

  The website for PyTorch
  Jupyter Notebook BSD-3-Clause 161 125 58 6 Updated May 13, 2021

• tutorials

  PyTorch tutorials.
  Python BSD-3-Clause 2,775 5,038 246 38 Updated May 13, 2021

• ignite

  High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

  python machine-learning deep-learning neural-network metrics pytorch
  Python BSD-3-Clause 460 3,553 89 (29 issues need help) 27 Updated May 13, 2021

• builder

  Continuous builder and binary build scripts for pytorch
  Shell BSD-2-Clause 124 167 21 20 Updated May 13, 2021

• test-infra

  This repository hosts code that supports the testing infrastructure for the main PyTorch repo. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.
  Python 2 3 0 0 Updated May 13, 2021

• audio

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

  audio python mp3 speech wav io
  Python BSD-2-Clause 315 1,306 117 (11 issues need help) 32 Updated May 13, 2021

• nestedtensor

  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  Jupyter Notebook BSD-3-Clause 18 131 36 5 Updated May 13, 2021

• elastic

  PyTorch elastic training
  Python BSD-3-Clause 74 641 6 2 Updated May 13, 2021

• vision

  Datasets, Transforms and Models specific to Computer Vision

  machine-learning computer-vision
  Python BSD-3-Clause 4,666 9,047 342 (51 issues need help) 127 Updated May 13, 2021

• xla

  Enabling PyTorch on Google TPU
  C++ 194 1,303 101 15 Updated May 13, 2021

• tensorpipe

  A tensor-aware point-to-point communication primitive for machine learning
  C++ 39 123 26 13 Updated May 13, 2021

• pytorch_sphinx_theme

  PyTorch Sphinx Theme
  CSS MIT 62 44 9 7 Updated May 13, 2021

• opacus

  Training PyTorch models with differential privacy

  machine-learning deep-learning neural-network differential-privacy privacy-preserving-machine-learning
  Python Apache-2.0 123 810 11 (2 issues need help) 4 Updated May 13, 2021

• csprng

  Cryptographically secure pseudorandom number generators for PyTorch
  Batchfile BSD-3-Clause 11 61 14 (6 issues need help) 7 Updated May 13, 2021

• cppdocs

  PyTorch C++ API Documentation
  HTML 28 114 6 3 Updated May 13, 2021

• ossci-job-dsl

  Jenkins job definitions for OSSCI
  Groovy Apache-2.0 38 7 0 3 Updated May 13, 2021

• benchmark
  Python BSD-3-Clause 73 262 54 33 Updated May 13, 2021

• fairseq

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

  python pytorch artificial-intelligence
  Python MIT 3,114 12,047 766 (100 issues need help) 68 Updated May 12, 2021

• captum

  Model interpretability and understanding for PyTorch

  interpretability feature-importance interpretable-ai interpretable-ml feature-attribution
  Python BSD-3-Clause 246 2,340 46 (1 issue needs help) 11 Updated May 12, 2021

• FBGEMM

  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  C++ 195 633 7 17 Updated May 12, 2021

• botorch

  Bayesian optimization in PyTorch
  Python MIT 203 1,937 37 11 Updated May 12, 2021

• android-demo-app

  PyTorch android examples of usage in applications
  Java 193 419 48 12 Updated May 12, 2021

• rfcs

  PyTorch RFCs (experimental)
  15 37 0 11 Updated May 11, 2021

• examples

  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  Python BSD-3-Clause 7,536 16,146 237 76 Updated May 11, 2021

• pytorch-probot

  PyTorch GitHub bot written in probot
  TypeScript ISC 4 7 1 3 Updated May 10, 2021