pytorch

Repositories

• tutorials

  PyTorch tutorials.
  Python BSD-3-Clause 2,739 4,992 253 70 Updated Apr 24, 2021

• pytorch

  Tensors and Dynamic neural networks in Python with strong GPU acceleration

  python machine-learning deep-learning neural-network gpu numpy autograd
  C++ 12,761 47,779 5,000+ (16 issues need help) 2,667 Updated Apr 24, 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 456 3,531 96 (33 issues need help) 32 Updated Apr 24, 2021

• captum

  Model interpretability and understanding for PyTorch

  interpretability feature-importance interpretable-ai interpretable-ml feature-attribution
  Python BSD-3-Clause 235 2,292 45 (1 issue needs help) 7 Updated Apr 24, 2021

• audio

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

  audio python mp3 speech wav io
  Python BSD-2-Clause 306 1,298 111 (12 issues need help) 30 Updated Apr 24, 2021

• text

  Data loaders and abstractions for text and NLP

  nlp deep-learning pytorch dataset data-loader
  Python BSD-3-Clause 626 2,735 227 (15 issues need help) 77 Updated Apr 24, 2021

• csprng

  Cryptographically secure pseudorandom number generators for PyTorch
  Batchfile BSD-3-Clause 11 58 14 (6 issues need help) 7 Updated Apr 24, 2021

• vision

  Datasets, Transforms and Models specific to Computer Vision

  machine-learning computer-vision
  Python BSD-3-Clause 4,594 8,914 329 (52 issues need help) 123 Updated Apr 24, 2021

• botorch

  Bayesian optimization in PyTorch
  Python MIT 202 1,922 36 11 Updated Apr 24, 2021

• cppdocs

  PyTorch C++ API Documentation
  HTML 26 111 6 3 Updated Apr 24, 2021

• pytorch.github.io

  The website for PyTorch
  Jupyter Notebook BSD-3-Clause 159 124 53 8 Updated Apr 24, 2021

• serve

  Model Serving on PyTorch
  Java Apache-2.0 270 1,719 101 (4 issues need help) 38 Updated Apr 24, 2021

• glow

  Compiler for Neural Network hardware accelerators
  C++ Apache-2.0 544 2,453 255 (2 issues need help) 95 Updated Apr 24, 2021

• kineto

  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  HTML 24 82 26 7 Updated Apr 24, 2021

• builder

  Continuous builder and binary build scripts for pytorch
  Shell BSD-2-Clause 120 163 20 22 Updated Apr 24, 2021

• xla

  Enabling PyTorch on Google TPU
  C++ 189 1,286 91 16 Updated Apr 24, 2021

• test-infra

  This repository will host code that will support the testing infrastructure for the current open source PyTorch repo. For example, this repo can host the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.
  Python 1 2 0 0 Updated Apr 24, 2021

• benchmark
  Python BSD-3-Clause 70 257 54 33 Updated Apr 24, 2021

• hydra-torch

  Configuration classes enabling type-safe PyTorch configuration for Hydra apps
  Python MIT 10 99 19 8 Updated Apr 23, 2021

• ossci-job-dsl

  Jenkins job definitions for OSSCI
  Groovy Apache-2.0 38 7 0 2 Updated Apr 23, 2021

• hub

  Submission to https://pytorch.org/hub/
  Python 152 1,020 15 3 Updated Apr 24, 2021

• tensorpipe

  A tensor-aware point-to-point communication primitive for machine learning
  C++ 37 119 28 9 Updated Apr 24, 2021

• fairseq

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

  python pytorch artificial-intelligence
  Python MIT 3,056 11,907 740 (97 issues need help) 67 Updated Apr 24, 2021

• translate

  Translate - a PyTorch Language Library

  machine-learning pytorch artificial-intelligence onnx
  Python BSD-3-Clause 182 689 11 16 Updated Apr 23, 2021

• elastic

  PyTorch elastic training
  Python BSD-3-Clause 73 637 6 1 Updated Apr 22, 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 117 794 11 (2 issues need help) 3 Updated Apr 22, 2021

• FBGEMM

  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  C++ 191 627 7 17 Updated Apr 22, 2021

• pytorch_sphinx_theme

  PyTorch Sphinx Theme
  CSS MIT 61 43 8 5 Updated Apr 21, 2021

• android-demo-app

  PyTorch android examples of usage in applications
  Java 182 395 45 11 Updated Apr 18, 2021

• nestedtensor

  [Prototype] Tools for the concurrent manipulation of variably sized Tensors.
  Jupyter Notebook BSD-3-Clause 17 129 44 7 Updated Apr 17, 2021