Skip to content
@pytorch

pytorch

Pinned

  1. pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    C++ 61.4k 17.1k

  2. examples Public

    A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

    Python 19.5k 9k

  3. tutorials Public

    PyTorch tutorials.

    Python 6.6k 3.6k

  4. vision Public

    Datasets, Transforms and Models specific to Computer Vision

    Python 13.1k 6.4k

  5. text Public

    Data loaders and abstractions for text and NLP

    Python 3.2k 768

  6. extension-cpp Public

    C++ extensions in PyTorch

    Python 802 169

Repositories

  • pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    C++ 61,416 17,091 5,000+ (55 issues need help) 801 Updated Jan 5, 2023
  • torchrec Public

    Pytorch domain library for recommendation systems

    Python 1,277 BSD-3-Clause 211 8 52 Updated Jan 5, 2023
  • benchmark Public

    TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

    Python 490 BSD-3-Clause 157 57 47 Updated Jan 5, 2023
  • torcheval Public

    A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to facilitate metric computation in distributed training and tools for PyTorch model evaluations.

    Python 84 25 1 1 Updated Jan 5, 2023
  • botorch Public

    Bayesian optimization in PyTorch

    Jupyter Notebook 2,483 MIT 296 61 17 Updated Jan 5, 2023
  • kineto Public

    A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.

    HTML 377 103 30 14 Updated Jan 5, 2023
  • audio Public

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

    Python 1,937 BSD-2-Clause 504 148 (12 issues need help) 58 Updated Jan 5, 2023
  • test-infra Public

    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.

    TypeScript 31 32 57 52 Updated Jan 5, 2023
  • xla Public

    Enabling PyTorch on Google TPU

    C++ 1,817 296 197 58 Updated Jan 5, 2023
  • multipy Public

    torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters in a single C++ process.

    C++ 93 25 32 (3 issues need help) 14 Updated Jan 5, 2023
View all repositories