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,361 49,248 5,000+ (22 issues need help) 2,825 Updated Jul 5, 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 473 3,603 95 (30 issues need help) 26 Updated Jul 4, 2021

• captum

  Model interpretability and understanding for PyTorch

  interpretability feature-importance interpretable-ai interpretable-ml feature-attribution
  Python BSD-3-Clause 258 2,457 54 (1 issue needs help) 8 Updated Jul 4, 2021

• text

  Data loaders and abstractions for text and NLP

  nlp deep-learning pytorch dataset data-loader
  Python BSD-3-Clause 650 2,798 240 (15 issues need help) 80 Updated Jul 4, 2021

• csprng

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

• vision

  Datasets, Transforms and Models specific to Computer Vision

  machine-learning computer-vision
  Python BSD-3-Clause 4,840 9,376 352 (52 issues need help) 123 Updated Jul 4, 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,358 109 (12 issues need help) 42 Updated Jul 4, 2021

• nestedtensor

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

• FBGEMM

  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  C++ 197 654 10 17 Updated Jul 4, 2021

• cppdocs

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

• pytorch.github.io

  The website for PyTorch
  Jupyter Notebook BSD-3-Clause 164 128 61 4 Updated Jul 4, 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 160 528 11 11 Updated Jul 4, 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 Jul 3, 2021

• botorch

  Bayesian optimization in PyTorch
  Python MIT 211 1,988 37 5 Updated Jul 3, 2021

• serve

  Model Serving on PyTorch
  Java Apache-2.0 329 1,932 92 (3 issues need help) 29 Updated Jul 3, 2021

• ort

  Accelerate PyTorch models with ONNX Runtime
  Python MIT 14 42 2 2 Updated Jul 2, 2021

• xla

  Enabling PyTorch on Google TPU
  C++ 206 1,357 62 15 Updated Jul 3, 2021

• glow

  Compiler for Neural Network hardware accelerators
  C++ Apache-2.0 562 2,530 260 (2 issues need help) 77 Updated Jul 3, 2021

• tutorials

  PyTorch tutorials.
  Python BSD-3-Clause 2,868 5,185 155 43 Updated Jul 2, 2021

• pytorch-ci-hud

  Better front page for ci.pytorch.org than Jenkins provides
  JavaScript MIT 20 6 1 1 Updated Jul 2, 2021

• fairseq

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

  python pytorch artificial-intelligence
  Python MIT 3,344 13,222 839 (104 issues need help) 28 Updated Jul 1, 2021

• rfcs

  PyTorch RFCs (experimental)
  16 38 1 12 Updated Jul 1, 2021

• elastic

  PyTorch elastic training
  Python BSD-3-Clause 78 654 4 2 Updated Jul 1, 2021

• hub

  Submission to https://pytorch.org/hub/
  Python 156 1,044 17 3 Updated Jul 1, 2021

• pytorch_sphinx_theme

  PyTorch Sphinx Theme
  CSS MIT 61 50 14 7 Updated Jul 1, 2021

• builder

  Continuous builder and binary build scripts for pytorch
  Shell BSD-2-Clause 128 175 22 18 Updated Jul 1, 2021

• tensorpipe

  A tensor-aware point-to-point communication primitive for machine learning
  C++ 41 131 27 15 Updated Jul 1, 2021

• kineto

  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  HTML 34 119 26 4 Updated Jul 1, 2021

• benchmark
  Python BSD-3-Clause 81 294 53 32 Updated Jul 1, 2021

• pytorch-probot

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