Skip to content
@pytorch

pytorch

Pinned Loading

  1. pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 90.5k 24.3k

  2. examples Public

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

    Python 23.1k 9.7k

  3. tutorials Public

    PyTorch tutorials.

    Jupyter Notebook 8.6k 4.2k

  4. vision Public

    Datasets, Transforms and Models specific to Computer Vision

    Python 16.9k 7.1k

  5. torchrec Public

    Pytorch domain library for recommendation systems

    Python 2.2k 522

Repositories

Showing 10 of 82 repositories
  • pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 90,470 24,308 5,000+ (89 issues need help) 1,263 Updated Jun 2, 2025
  • 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.

    TypeScript 92 97 228 (1 issue needs help) 201 Updated Jun 2, 2025
  • torchtitan Public

    A PyTorch native platform for training generative AI models

    Python 3,873 BSD-3-Clause 379 100 (1 issue needs help) 69 Updated Jun 2, 2025
  • executorch Public

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

    Python 2,905 571 491 (33 issues need help) 552 Updated Jun 1, 2025
  • benchmark Public

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

    Python 946 BSD-3-Clause 309 79 74 Updated Jun 2, 2025
  • xla Public

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

    Python 2,613 530 643 (49 issues need help) 188 Updated Jun 2, 2025
  • ao Public

    PyTorch native quantization and sparsity for training and inference

    Python 2,073 BSD-3-Clause 273 246 (2 issues need help) 130 Updated Jun 2, 2025
  • docs Public

    This repository is automatically generated to contain the website source for the PyTorch documentation at https//pytorch.org/docs.

    HTML 9 BSD-3-Clause 4 2 0 Updated Jun 2, 2025
  • functorch Public

    functorch is JAX-like composable function transforms for PyTorch.

    Jupyter Notebook 1,429 BSD-3-Clause 104 233 (1 issue needs help) 17 Updated Jun 1, 2025
  • TensorRT Public

    PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

    Python 2,767 BSD-3-Clause 364 208 24 Updated Jun 1, 2025
View all repositories