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,952 48,262 5,000+ (23 issues need help) 2,681 Updated May 18, 2021

• tensorpipe

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

• cppdocs

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

• vision

  Datasets, Transforms and Models specific to Computer Vision

  machine-learning computer-vision
  Python BSD-3-Clause 4,672 9,072 344 (52 issues need help) 124 Updated May 18, 2021

• pytorch.github.io

  The website for PyTorch
  Jupyter Notebook BSD-3-Clause 162 126 58 6 Updated May 18, 2021

• kineto

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

• serve

  Model Serving on PyTorch
  Java Apache-2.0 292 1,788 93 (3 issues need help) 25 Updated May 18, 2021

• captum

  Model interpretability and understanding for PyTorch

  interpretability feature-importance interpretable-ai interpretable-ml feature-attribution
  Python BSD-3-Clause 247 2,351 47 (1 issue needs help) 10 Updated May 18, 2021

• botorch

  Bayesian optimization in PyTorch
  Python MIT 204 1,941 38 10 Updated May 18, 2021

• glow

  Compiler for Neural Network hardware accelerators
  C++ Apache-2.0 551 2,472 254 (2 issues need help) 98 Updated May 18, 2021

• tutorials

  PyTorch tutorials.
  Python BSD-3-Clause 2,780 5,044 247 39 Updated May 18, 2021

• hub

  Submission to https://pytorch.org/hub/
  Python 154 1,031 15 4 Updated May 17, 2021

• nestedtensor

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

• elastic

  PyTorch elastic training
  Python BSD-3-Clause 74 643 2 1 Updated May 18, 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 461 3,561 89 (28 issues need help) 26 Updated May 18, 2021

• xla

  Enabling PyTorch on Google TPU
  C++ 195 1,306 99 15 Updated May 18, 2021

• builder

  Continuous builder and binary build scripts for pytorch
  Shell BSD-2-Clause 124 167 21 21 Updated May 18, 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,310 120 (12 issues need help) 34 Updated May 17, 2021

• text

  Data loaders and abstractions for text and NLP

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

• benchmark
  Python BSD-3-Clause 73 263 54 32 Updated May 17, 2021

• fairseq

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

  python pytorch artificial-intelligence
  Python MIT 3,121 12,084 771 (100 issues need help) 70 Updated May 17, 2021

• csprng

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

• opacus

  Training PyTorch models with differential privacy

  machine-learning deep-learning neural-network differential-privacy privacy-preserving-machine-learning
  Python Apache-2.0 124 811 10 (2 issues need help) 2 Updated May 17, 2021

• FBGEMM

  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  C++ 195 635 7 17 Updated May 17, 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 16, 2021

• ort

  Accelerate PyTorch models with ONNX Runtime
  Shell MIT 1 8 0 0 Updated May 14, 2021

• pytorch_sphinx_theme

  PyTorch Sphinx Theme
  CSS MIT 62 44 10 6 Updated May 15, 2021

• ossci-job-dsl

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

• android-demo-app

  PyTorch android examples of usage in applications
  Java 192 424 48 12 Updated May 12, 2021

• rfcs

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