pytorch

Repositories

• vision

  Datasets, Transforms and Models specific to Computer Vision

  machine-learning computer-vision
  Python BSD-3-Clause 4,821 9,340 348 (52 issues need help) 119 Updated Jun 29, 2021

• text

  Data loaders and abstractions for text and NLP

  nlp deep-learning pytorch dataset data-loader
  Python BSD-3-Clause 648 2,797 238 (15 issues need help) 80 Updated Jun 29, 2021

• nestedtensor

  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  Jupyter Notebook BSD-3-Clause 19 144 38 3 Updated Jun 29, 2021

• audio

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

  audio python mp3 speech wav io
  Python BSD-2-Clause 330 1,356 109 (12 issues need help) 41 Updated Jun 29, 2021

• csprng

  Cryptographically secure pseudorandom number generators for PyTorch
  Batchfile BSD-3-Clause 11 63 17 (6 issues need help) 7 Updated Jun 29, 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,295 49,142 5,000+ (22 issues need help) 2,845 Updated Jun 29, 2021

• tensorpipe

  A tensor-aware point-to-point communication primitive for machine learning
  C++ 41 130 27 15 Updated Jun 29, 2021

• cppdocs

  PyTorch C++ API Documentation
  HTML 28 115 6 3 Updated Jun 29, 2021

• kineto

  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  HTML 34 117 27 4 Updated Jun 29, 2021

• pytorch.github.io

  The website for PyTorch
  Jupyter Notebook BSD-3-Clause 163 128 61 4 Updated Jun 29, 2021

• tutorials

  PyTorch tutorials.
  Python BSD-3-Clause 2,860 5,170 154 42 Updated Jun 29, 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 471 3,600 97 (29 issues need help) 26 Updated Jun 29, 2021

• glow

  Compiler for Neural Network hardware accelerators
  C++ Apache-2.0 560 2,529 260 (2 issues need help) 106 Updated Jun 29, 2021

• FBGEMM

  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  C++ 198 653 10 21 Updated Jun 29, 2021

• fairseq

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

  python pytorch artificial-intelligence
  Python MIT 3,317 13,180 828 (102 issues need help) 28 Updated Jun 29, 2021

• botorch

  Bayesian optimization in PyTorch
  Python MIT 209 1,981 38 5 Updated Jun 29, 2021

• captum

  Model interpretability and understanding for PyTorch

  interpretability feature-importance interpretable-ai interpretable-ml feature-attribution
  Python BSD-3-Clause 258 2,452 52 (1 issue needs help) 9 Updated Jun 29, 2021

• ort

  Accelerate PyTorch models with ONNX Runtime
  Python MIT 13 40 2 4 Updated Jun 29, 2021

• benchmark
  Python BSD-3-Clause 79 272 53 31 Updated Jun 29, 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 4 21 7 1 Updated Jun 29, 2021

• serve

  Model Serving on PyTorch
  Java Apache-2.0 327 1,921 89 (3 issues need help) 28 Updated Jun 28, 2021

• xla

  Enabling PyTorch on Google TPU
  C++ 203 1,355 60 12 Updated Jun 29, 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 6 5 0 3 Updated Jun 28, 2021

• opacus

  Training PyTorch models with differential privacy

  machine-learning deep-learning neural-network differential-privacy privacy-preserving-machine-learning
  Python Apache-2.0 129 836 12 (3 issues need help) 3 Updated Jun 28, 2021

• pytorch-ci-hud

  Better front page for ci.pytorch.org than Jenkins provides
  JavaScript MIT 20 4 1 3 Updated Jun 28, 2021

• builder

  Continuous builder and binary build scripts for pytorch
  Shell BSD-2-Clause 127 172 22 18 Updated Jun 28, 2021

• 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 157 526 10 9 Updated Jun 25, 2021

• pytorch-probot

  PyTorch GitHub bot written in probot
  TypeScript ISC 5 8 2 4 Updated Jun 23, 2021

• elastic

  PyTorch elastic training
  Python BSD-3-Clause 78 654 4 1 Updated Jun 24, 2021

• android-demo-app

  PyTorch android examples of usage in applications
  Java 231 518 63 15 Updated Jun 22, 2021