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,275 49,078 5,000+ (22 issues need help) 2,836 Updated Jun 26, 2021

• pytorch.github.io

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

• text

  Data loaders and abstractions for text and NLP

  nlp deep-learning pytorch dataset data-loader
  Python BSD-3-Clause 647 2,795 238 (15 issues need help) 79 Updated Jun 26, 2021

• xla

  Enabling PyTorch on Google TPU
  C++ 203 1,353 60 13 Updated Jun 25, 2021

• fairseq

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

  python pytorch artificial-intelligence
  Python MIT 3,308 13,154 823 (102 issues need help) 28 Updated Jun 26, 2021

• tutorials

  PyTorch tutorials.
  Python BSD-3-Clause 2,860 5,164 154 40 Updated Jun 26, 2021

• serve

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

• audio

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

  audio python mp3 speech wav io
  Python BSD-2-Clause 331 1,357 109 (12 issues need help) 43 Updated Jun 26, 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 469 3,596 96 (29 issues need help) 26 Updated Jun 25, 2021

• glow

  Compiler for Neural Network hardware accelerators
  C++ Apache-2.0 557 2,525 259 (2 issues need help) 107 Updated Jun 25, 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 0 Updated Jun 25, 2021

• vision

  Datasets, Transforms and Models specific to Computer Vision

  machine-learning computer-vision
  Python BSD-3-Clause 4,804 9,322 345 (52 issues need help) 121 Updated Jun 25, 2021

• captum

  Model interpretability and understanding for PyTorch

  interpretability feature-importance interpretable-ai interpretable-ml feature-attribution
  Python BSD-3-Clause 258 2,447 52 (1 issue needs help) 10 Updated Jun 25, 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 833 12 (3 issues need help) 2 Updated Jun 25, 2021

• botorch

  Bayesian optimization in PyTorch
  Python MIT 208 1,975 39 9 Updated Jun 25, 2021

• benchmark
  Python BSD-3-Clause 79 272 53 31 Updated Jun 25, 2021

• builder

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

• nestedtensor

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

• kineto

  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  HTML 32 114 26 4 Updated Jun 25, 2021

• ort

  Accelerate PyTorch models with ONNX Runtime
  Python MIT 13 38 2 4 Updated Jun 25, 2021

• FBGEMM

  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  C++ 197 654 9 19 Updated Jun 25, 2021

• csprng

  Cryptographically secure pseudorandom number generators for PyTorch
  Batchfile BSD-3-Clause 11 63 17 (6 issues need help) 7 Updated Jun 25, 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 158 524 10 9 Updated Jun 25, 2021

• cppdocs

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

• pytorch-ci-hud

  Better front page for ci.pytorch.org than Jenkins provides
  JavaScript MIT 20 4 1 4 Updated Jun 25, 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 24, 2021

• tensorpipe

  A tensor-aware point-to-point communication primitive for machine learning
  C++ 41 129 27 16 Updated Jun 24, 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 23, 2021

• android-demo-app

  PyTorch android examples of usage in applications
  Java 230 513 64 15 Updated Jun 22, 2021