pytorch

Repositories

• pytorch.github.io

  The website for PyTorch
  Jupyter Notebook BSD-3-Clause 169 130 64 8 Updated Jul 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,476 49,779 5,000+ (22 issues need help) 2,840 Updated Jul 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 478 3,622 101 (34 issues need help) 28 Updated Jul 29, 2021

• elastic

  PyTorch elastic training
  Python BSD-3-Clause 80 663 4 2 Updated Jul 29, 2021

• vision

  Datasets, Transforms and Models specific to Computer Vision

  machine-learning computer-vision
  Python BSD-3-Clause 4,932 9,518 366 (52 issues need help) 128 Updated Jul 29, 2021

• audio

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

  audio python mp3 speech wav io
  Python BSD-2-Clause 340 1,378 103 (12 issues need help) 37 Updated Jul 29, 2021

• ort

  Accelerate PyTorch models with ONNX Runtime
  Python MIT 18 121 5 4 Updated Jul 29, 2021

• tutorials

  PyTorch tutorials.
  Python BSD-3-Clause 2,915 5,267 144 47 Updated Jul 29, 2021

• xla

  Enabling PyTorch on Google TPU
  C++ 211 1,386 63 11 Updated Jul 29, 2021

• pytorch-ci-hud

  Better front page for ci.pytorch.org than Jenkins provides
  JavaScript MIT 20 6 5 0 Updated Jul 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 7 3 5 Updated Jul 29, 2021

• serve

  Model Serving on PyTorch
  Java Apache-2.0 352 1,979 93 (3 issues need help) 22 Updated Jul 29, 2021

• FBGEMM

  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  C++ 198 670 10 20 Updated Jul 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 8 31 8 3 Updated Jul 29, 2021

• fairseq

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

  python pytorch artificial-intelligence
  Python MIT 3,414 13,411 878 (104 issues need help) 21 Updated Jul 28, 2021

• examples

  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  Python BSD-3-Clause 7,724 16,610 241 76 Updated Jul 29, 2021

• glow

  Compiler for Neural Network hardware accelerators
  C++ Apache-2.0 564 2,552 262 (2 issues need help) 27 Updated Jul 29, 2021

• kineto

  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  HTML 40 126 26 5 Updated Jul 28, 2021

• builder

  Continuous builder and binary build scripts for pytorch
  Shell BSD-2-Clause 135 177 23 22 Updated Jul 28, 2021

• botorch

  Bayesian optimization in PyTorch
  Python MIT 216 2,014 44 8 Updated Jul 29, 2021

• text

  Data loaders and abstractions for text and NLP

  nlp deep-learning pytorch dataset data-loader
  Python BSD-3-Clause 652 2,810 245 (15 issues need help) 78 Updated Jul 28, 2021

• tensorpipe

  A tensor-aware point-to-point communication primitive for machine learning
  C++ 40 134 27 15 Updated Jul 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 132 860 13 (3 issues need help) 4 Updated Jul 28, 2021

• nestedtensor

  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  Jupyter Notebook BSD-3-Clause 18 156 39 3 Updated Jul 28, 2021

• csprng

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

• cppdocs

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

• captum

  Model interpretability and understanding for PyTorch

  interpretability feature-importance interpretable-ai interpretable-ml feature-attribution
  Python BSD-3-Clause 267 2,502 54 (1 issue needs help) 11 Updated Jul 25, 2021

• java-demo
  Jupyter Notebook 15 71 2 0 Updated Jul 23, 2021

• hub

  Submission to https://pytorch.org/hub/
  Python 159 1,057 18 3 Updated Jul 22, 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
  C BSD-2-Clause 165 542 10 9 Updated Jul 20, 2021