Skip to content
@pytorch

pytorch

Pinned Loading

  1. pytorch pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 85.1k 22.9k

  2. examples examples Public

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

    Python 22.5k 9.6k

  3. tutorials tutorials Public

    PyTorch tutorials.

    Jupyter Notebook 8.3k 4.1k

  4. vision vision Public

    Datasets, Transforms and Models specific to Computer Vision

    Python 16.4k 7k

  5. torchrec torchrec Public

    Pytorch domain library for recommendation systems

    Python 2k 450

Repositories

Showing 10 of 80 repositories
  • pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    pytorch/pytorch’s past year of commit activity
  • benchmark Public

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

    pytorch/benchmark’s past year of commit activity
    Python 885 BSD-3-Clause 291 67 54 Updated Dec 18, 2024
  • executorch Public

    On-device AI across mobile, embedded and edge for PyTorch

    pytorch/executorch’s past year of commit activity
    C++ 2,306 398 249 (2 issues need help) 627 Updated Dec 18, 2024
  • TensorRT Public

    PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

    pytorch/TensorRT’s past year of commit activity
    Python 2,624 BSD-3-Clause 352 188 16 Updated Dec 18, 2024
  • xla Public

    Enabling PyTorch on XLA Devices (e.g. Google TPU)

    pytorch/xla’s past year of commit activity
    C++ 2,502 486 593 (51 issues need help) 149 Updated Dec 18, 2024
  • FBGEMM Public

    FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

    pytorch/FBGEMM’s past year of commit activity
    C++ 1,222 516 39 380 Updated Dec 18, 2024
  • ao Public

    PyTorch native quantization and sparsity for training and inference

    pytorch/ao’s past year of commit activity
    Python 1,675 BSD-3-Clause 185 185 (2 issues need help) 86 Updated Dec 18, 2024
  • test-infra Public

    This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.

    pytorch/test-infra’s past year of commit activity
    TypeScript 83 83 217 (1 issue needs help) 166 Updated Dec 18, 2024
  • cppdocs Public

    PyTorch C++ API Documentation

    pytorch/cppdocs’s past year of commit activity
    HTML 212 34 7 3 Updated Dec 18, 2024
  • torchrec Public

    Pytorch domain library for recommendation systems

    pytorch/torchrec’s past year of commit activity
    Python 1,976 BSD-3-Clause 450 63 272 Updated Dec 17, 2024