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++ 13,061 48,572 5,000+ (23 issues need help) 2,701 Updated Jun 2, 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,576 97 (31 issues need help) 29 Updated Jun 2, 2021

• ossci-job-dsl

  Jenkins job definitions for OSSCI
  Groovy Apache-2.0 37 7 0 3 Updated Jun 2, 2021

• serve

  Model Serving on PyTorch
  Java Apache-2.0 308 1,856 92 (3 issues need help) 25 Updated Jun 2, 2021

• botorch

  Bayesian optimization in PyTorch
  Python MIT 205 1,951 39 10 Updated Jun 2, 2021

• glow

  Compiler for Neural Network hardware accelerators
  C++ Apache-2.0 553 2,481 255 (2 issues need help) 99 Updated Jun 2, 2021

• kineto

  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  HTML 27 104 22 7 Updated Jun 2, 2021

• vision

  Datasets, Transforms and Models specific to Computer Vision

  machine-learning computer-vision
  Python BSD-3-Clause 4,718 9,168 342 (53 issues need help) 124 Updated Jun 2, 2021

• benchmark
  Python BSD-3-Clause 74 268 54 36 Updated Jun 2, 2021

• builder

  Continuous builder and binary build scripts for pytorch
  Shell BSD-2-Clause 125 169 21 19 Updated Jun 2, 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 4 5 0 1 Updated Jun 2, 2021

• audio

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

  audio python mp3 speech wav io
  Python BSD-2-Clause 320 1,330 117 (10 issues need help) 34 Updated Jun 3, 2021

• xla

  Enabling PyTorch on Google TPU
  C++ 199 1,331 102 13 Updated Jun 2, 2021

• tutorials

  PyTorch tutorials.
  Python BSD-3-Clause 2,813 5,085 245 45 Updated Jun 2, 2021

• FBGEMM

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

• captum

  Model interpretability and understanding for PyTorch

  interpretability feature-importance interpretable-ai interpretable-ml feature-attribution
  Python BSD-3-Clause 248 2,398 46 (1 issue needs help) 10 Updated Jun 2, 2021

• pytorch.github.io

  The website for PyTorch
  Jupyter Notebook BSD-3-Clause 162 126 59 8 Updated Jun 2, 2021

• hub

  Submission to https://pytorch.org/hub/
  Python 156 1,040 15 1 Updated Jun 2, 2021

• tensorpipe

  A tensor-aware point-to-point communication primitive for machine learning
  C++ 40 126 26 15 Updated Jun 2, 2021

• text

  Data loaders and abstractions for text and NLP

  nlp deep-learning pytorch dataset data-loader
  Python BSD-3-Clause 635 2,773 233 (15 issues need help) 76 Updated Jun 2, 2021

• opacus

  Training PyTorch models with differential privacy

  machine-learning deep-learning neural-network differential-privacy privacy-preserving-machine-learning
  Python Apache-2.0 126 821 10 (2 issues need help) 1 Updated Jun 2, 2021

• nestedtensor

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

• csprng

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

• cppdocs

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

• ort

  Accelerate PyTorch models with ONNX Runtime
  Python MIT 6 13 1 1 Updated Jun 2, 2021

• fairseq

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

  python pytorch artificial-intelligence
  Python MIT 3,230 12,849 786 (100 issues need help) 71 Updated Jun 1, 2021

• pytorch-ci-hud

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

• android-demo-app

  PyTorch android examples of usage in applications
  Java 205 449 52 14 Updated May 27, 2021

• examples

  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  Python BSD-3-Clause 7,593 16,293 239 76 Updated May 27, 2021

• pytorch-probot

  PyTorch GitHub bot written in probot
  TypeScript ISC 5 7 1 3 Updated May 27, 2021