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++ 12,407 46,733 5,000+ (18 issues need help) 2,408 Updated Mar 8, 2021

• vision

  Datasets, Transforms and Models specific to Computer Vision

  machine-learning computer-vision
  Python BSD-3-Clause 4,388 8,536 329 (55 issues need help) 132 Updated Mar 8, 2021

• text

  Data loaders and abstractions for text and NLP

  nlp deep-learning pytorch dataset data-loader
  Python BSD-3-Clause 616 2,669 219 (15 issues need help) 72 Updated Mar 8, 2021

• opacus

  Training PyTorch models with differential privacy

  machine-learning deep-learning neural-network differential-privacy privacy-preserving-machine-learning
  Jupyter Notebook Apache-2.0 101 754 10 (2 issues need help) 5 Updated Mar 8, 2021

• audio

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

  audio python mp3 speech wav io
  Python BSD-2-Clause 287 1,251 108 (11 issues need help) 38 Updated Mar 8, 2021

• fairseq

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

  python pytorch artificial-intelligence
  Python MIT 2,918 11,430 631 (96 issues need help) 56 Updated Mar 8, 2021

• FBGEMM

  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  C++ 174 590 9 21 Updated Mar 8, 2021

• botorch

  Bayesian optimization in PyTorch
  Python MIT 186 1,859 34 13 Updated Mar 8, 2021

• benchmark
  Python BSD-3-Clause 63 237 43 22 Updated Mar 8, 2021

• csprng

  Cryptographically secure pseudorandom number generators for PyTorch
  Batchfile BSD-3-Clause 7 49 15 (6 issues need help) 6 Updated Mar 8, 2021

• pytorch.github.io

  The website for PyTorch
  Jupyter Notebook BSD-3-Clause 145 119 45 3 Updated Mar 8, 2021

• tutorials

  PyTorch tutorials.
  Python BSD-3-Clause 2,621 4,785 243 177 Updated Mar 8, 2021

• tensorpipe

  A tensor-aware point-to-point communication primitive for machine learning
  C++ 25 107 27 6 Updated Mar 8, 2021

• kineto

  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  HTML 15 56 7 6 Updated Mar 8, 2021

• glow

  Compiler for Neural Network hardware accelerators
  C++ Apache-2.0 513 2,404 249 (2 issues need help) 84 Updated Mar 8, 2021

• hydra-torch

  Configuration classes enabling type-safe PyTorch configuration for Hydra apps
  Python MIT 6 88 21 7 Updated Mar 8, 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 438 3,450 81 (28 issues need help) 28 Updated Mar 8, 2021

• xla

  Enabling PyTorch on Google TPU
  C++ 179 1,229 65 7 Updated Mar 8, 2021

• examples

  A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
  Python BSD-3-Clause 7,300 15,598 230 74 Updated Mar 8, 2021

• serve

  Model Serving on PyTorch
  Java Apache-2.0 244 1,618 97 (1 issue needs help) 58 Updated Mar 7, 2021

• builder

  Continuous builder and binary build scripts for pytorch
  Shell BSD-2-Clause 107 159 17 20 Updated Mar 6, 2021

• android-demo-app

  PyTorch android examples of usage in applications
  Java 155 327 32 11 Updated Mar 6, 2021

• elastic

  PyTorch elastic training
  Python BSD-3-Clause 64 616 7 0 Updated Mar 6, 2021

• pytorch_sphinx_theme

  PyTorch Sphinx Theme
  CSS MIT 51 38 5 4 Updated Mar 5, 2021

• cppdocs

  PyTorch C++ API Documentation
  HTML 21 109 5 3 Updated Mar 5, 2021

• translate

  Translate - a PyTorch Language Library

  machine-learning pytorch artificial-intelligence onnx
  Python BSD-3-Clause 172 681 10 16 Updated Mar 4, 2021

• rfcs

  PyTorch RFCs (experimental)
  9 29 0 9 Updated Mar 3, 2021

• hub

  Submission to https://pytorch.org/hub/
  Python 138 1,009 14 3 Updated Mar 2, 2021

• nestedtensor

  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  Jupyter Notebook BSD-3-Clause 16 115 43 6 Updated Feb 28, 2021

• captum

  Model interpretability and understanding for PyTorch

  interpretability feature-importance interpretable-ai interpretable-ml feature-attribution
  Python BSD-3-Clause 221 2,194 40 (1 issue needs help) 7 Updated Feb 26, 2021