pytorch

Repositories

• glow

  Compiler for Neural Network hardware accelerators
  C++ Apache-2.0 495 2,318 240 (2 issues need help) 49 Updated Dec 2, 2020

• pytorch

  Tensors and Dynamic neural networks in Python with strong GPU acceleration

  python machine-learning deep-learning neural-network gpu numpy autograd
  C++ 11,715 44,367 5,000+ (24 issues need help) 2,202 Updated Dec 2, 2020

• serve

  Model Serving on PyTorch
  Java Apache-2.0 201 1,396 95 (1 issue needs help) 34 Updated Dec 2, 2020

• nestedtensor

  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  Jupyter Notebook BSD-3-Clause 13 98 41 4 Updated Dec 2, 2020

• botorch

  Bayesian optimization in PyTorch
  Python MIT 170 1,764 32 13 Updated Dec 2, 2020

• tutorials

  PyTorch tutorials.
  Jupyter Notebook BSD-3-Clause 2,444 4,459 217 181 Updated Dec 1, 2020

• benchmark
  Python BSD-3-Clause 54 218 35 9 Updated Dec 2, 2020

• fairseq

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

  python pytorch artificial-intelligence
  Python MIT 2,648 10,530 494 (88 issues need help) 50 Updated Dec 2, 2020

• vision

  Datasets, Transforms and Models specific to Computer Vision

  machine-learning computer-vision
  Python BSD-3-Clause 4,048 7,837 332 (65 issues need help) 132 Updated Dec 2, 2020

• builder

  Continuous builder and binary build scripts for pytorch
  Shell BSD-2-Clause 103 144 15 22 Updated Dec 2, 2020

• xla

  Enabling PyTorch on Google TPU
  C++ 159 1,120 59 3 Updated Dec 2, 2020

• audio

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

  audio python mp3 speech wav io
  Python BSD-2-Clause 261 1,145 110 (13 issues need help) 49 Updated Dec 1, 2020

• text

  Data loaders and abstractions for text and NLP

  nlp deep-learning pytorch dataset data-loader
  Python BSD-3-Clause 583 2,570 199 (15 issues need help) 59 Updated Dec 1, 2020

• hub

  Submission to https://pytorch.org/hub/
  Python 128 981 12 2 Updated Dec 1, 2020

• dr-ci

  Diagnose and remediate CI jobs
  Haskell MIT 2 2 2 3 Updated Dec 1, 2020

• hydra-torch

  Configuration classes enabling type-safe PyTorch configuration for Hydra apps
  Python MIT 2 29 11 3 Updated Dec 1, 2020

• pytorch.github.io

  The website for PyTorch
  Jupyter Notebook BSD-3-Clause 136 107 37 8 Updated Dec 1, 2020

• captum

  Model interpretability and understanding for PyTorch

  interpretability feature-importance interpretable-ai interpretable-ml feature-attribution
  Python BSD-3-Clause 186 1,932 40 (1 issue needs help) 10 Updated Dec 1, 2020

• rfcs

  PyTorch RFCs (experimental)
  4 17 0 5 Updated Dec 1, 2020

• tensorpipe

  A tensor-aware point-to-point communication primitive for machine learning
  C++ 16 81 23 5 Updated Dec 1, 2020

• csprng

  Cryptographically secure pseudorandom number generators for PyTorch
  Batchfile BSD-3-Clause 5 32 12 (4 issues need help) 5 Updated Dec 1, 2020

• cppdocs

  PyTorch C++ API Documentation
  HTML 20 104 4 3 Updated Dec 1, 2020

• kineto

  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  C++ BSD-3-Clause 1 8 0 4 Updated Nov 30, 2020

• 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 closember
  Python BSD-3-Clause 413 3,162 79 (27 issues need help) 26 Updated Nov 30, 2020

• examples

  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  Python BSD-3-Clause 6,957 14,823 214 66 Updated Nov 30, 2020

• opacus

  Training PyTorch models with differential privacy

  differential-privacy privacy-preserving-machine-learning dp-sgd
  Python Apache-2.0 86 661 4 3 Updated Nov 27, 2020

• pytorch_sphinx_theme

  PyTorch Sphinx Theme
  CSS MIT 45 36 2 15 Updated Nov 23, 2020

• 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 141 449 6 7 Updated Nov 21, 2020

• FBGEMM

  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  C++ 158 537 10 9 Updated Nov 19, 2020

• translate

  Translate - a PyTorch Language Library

  machine-learning pytorch artificial-intelligence onnx
  Python BSD-3-Clause 163 656 10 17 Updated Nov 16, 2020