pytorch

Repositories

• 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 463 3,589 96 (29 issues need help) 29 Updated Jun 12, 2021

• captum

  Model interpretability and understanding for PyTorch

  interpretability feature-importance interpretable-ai interpretable-ml feature-attribution
  Python BSD-3-Clause 251 2,417 48 (1 issue needs help) 11 Updated Jun 12, 2021

• pytorch

  Tensors and Dynamic neural networks in Python with strong GPU acceleration

  python machine-learning deep-learning neural-network gpu numpy autograd
  C++ 13,128 48,804 5,000+ (23 issues need help) 2,742 Updated Jun 12, 2021

• pytorch.github.io

  The website for PyTorch
  Jupyter Notebook BSD-3-Clause 164 128 62 7 Updated Jun 12, 2021

• hub

  Submission to https://pytorch.org/hub/
  Python 158 1,041 16 1 Updated Jun 12, 2021

• glow

  Compiler for Neural Network hardware accelerators
  C++ Apache-2.0 553 2,484 258 (2 issues need help) 106 Updated Jun 12, 2021

• text

  Data loaders and abstractions for text and NLP

  nlp deep-learning pytorch dataset data-loader
  Python BSD-3-Clause 640 2,786 235 (15 issues need help) 77 Updated Jun 12, 2021

• nestedtensor

  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  Jupyter Notebook BSD-3-Clause 18 138 38 2 Updated Jun 12, 2021

• csprng

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

• vision

  Datasets, Transforms and Models specific to Computer Vision

  machine-learning computer-vision
  Python BSD-3-Clause 4,756 9,237 343 (51 issues need help) 125 Updated Jun 12, 2021

• audio

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

  audio python mp3 speech wav io
  Python BSD-2-Clause 324 1,346 121 (12 issues need help) 34 Updated Jun 12, 2021

• FBGEMM

  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  C++ 196 645 7 19 Updated Jun 12, 2021

• xla

  Enabling PyTorch on Google TPU
  C++ 199 1,341 106 13 Updated Jun 12, 2021

• torchx

  TorchX is a library containing standard DSLs for authoring and running PyTorch related components for an E2E production ML pipeline
  Python BSD-3-Clause 0 2 0 1 Updated Jun 12, 2021

• serve

  Model Serving on PyTorch
  Java Apache-2.0 312 1,881 102 (3 issues need help) 22 Updated Jun 12, 2021

• benchmark
  Python BSD-3-Clause 77 269 53 35 Updated Jun 12, 2021

• fairseq

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

  python pytorch artificial-intelligence
  Python MIT 3,261 13,001 797 (100 issues need help) 45 Updated Jun 12, 2021

• cppdocs

  PyTorch C++ API Documentation
  HTML 28 116 6 3 Updated Jun 11, 2021

• builder

  Continuous builder and binary build scripts for pytorch
  Shell BSD-2-Clause 126 169 21 17 Updated Jun 11, 2021

• tensorpipe

  A tensor-aware point-to-point communication primitive for machine learning
  C++ 40 127 27 15 Updated Jun 11, 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 5 5 0 2 Updated Jun 11, 2021

• examples

  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  Python BSD-3-Clause 7,615 16,341 240 76 Updated Jun 11, 2021

• tutorials

  PyTorch tutorials.
  Python BSD-3-Clause 2,835 5,122 212 52 Updated Jun 11, 2021

• kineto

  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  HTML 29 106 21 8 Updated Jun 11, 2021

• botorch

  Bayesian optimization in PyTorch
  Python MIT 205 1,959 39 13 Updated Jun 11, 2021

• pytorch-probot

  PyTorch GitHub bot written in probot
  TypeScript ISC 5 7 1 2 Updated Jun 11, 2021

• pytorch-integration-testing

  Testing downstream libraries using pytorch release candidates
  Makefile MIT 8 3 0 0 Updated Jun 10, 2021

• pytorch-ci-hud

  Better front page for ci.pytorch.org than Jenkins provides
  JavaScript MIT 19 4 0 0 Updated Jun 10, 2021

• rfcs

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

• ios-demo-app

  PyTorch iOS examples
  Swift 65 223 11 2 Updated Jun 10, 2021