Skip to content
@pytorch

pytorch

Pinned Loading

  1. pytorch pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 85.7k 23.1k

  2. examples examples Public

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

    Python 22.6k 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.5k 7k

  5. torchrec torchrec Public

    Pytorch domain library for recommendation systems

    Python 2k 458

Repositories

Showing 10 of 81 repositories
  • pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    pytorch/pytorch’s past year of commit activity
  • 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 84 219 (1 issue needs help) 164 Updated Jan 11, 2025
  • benchmark Public

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

    pytorch/benchmark’s past year of commit activity
    Python 893 BSD-3-Clause 294 67 53 Updated Jan 11, 2025
  • executorch Public

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

    pytorch/executorch’s past year of commit activity
    C++ 2,384 415 274 (2 issues need help) 674 Updated Jan 11, 2025
  • torchcodec Public

    PyTorch video decoding

    pytorch/torchcodec’s past year of commit activity
    Python 204 BSD-3-Clause 18 37 11 Updated Jan 11, 2025
  • torchtune Public

    PyTorch native post-training library

    pytorch/torchtune’s past year of commit activity
    Python 4,676 BSD-3-Clause 484 248 (6 issues need help) 56 Updated Jan 11, 2025
  • 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,238 520 31 379 Updated Jan 11, 2025
  • TensorRT Public

    PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

    pytorch/TensorRT’s past year of commit activity
    Python 2,640 BSD-3-Clause 355 194 13 Updated Jan 11, 2025
  • torchrec Public

    Pytorch domain library for recommendation systems

    pytorch/torchrec’s past year of commit activity
    Python 2,005 BSD-3-Clause 458 60 273 Updated Jan 11, 2025
  • vision Public

    Datasets, Transforms and Models specific to Computer Vision

    pytorch/vision’s past year of commit activity
    Python 16,454 BSD-3-Clause 6,984 858 (49 issues need help) 228 Updated Jan 11, 2025