Skip to content
@pytorch

pytorch

Pinned Loading

  1. pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 89.3k 24k

  2. examples Public

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

    Python 23k 9.6k

  3. tutorials Public

    PyTorch tutorials.

    Jupyter Notebook 8.5k 4.1k

  4. vision Public

    Datasets, Transforms and Models specific to Computer Vision

    Python 16.8k 7k

  5. torchrec Public

    Pytorch domain library for recommendation systems

    Python 2.1k 492

Repositories

Showing 10 of 81 repositories
  • pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

  • ao Public

    PyTorch native quantization and sparsity for training and inference

    Python 1,987 BSD-3-Clause 247 231 (2 issues need help) 117 Updated Apr 24, 2025
  • TensorRT Public

    PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

    Python 2,731 BSD-3-Clause 362 208 24 Updated Apr 24, 2025
  • executorch Public

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

    Python 2,754 528 424 (14 issues need help) 580 Updated Apr 24, 2025
  • torchx Public

    TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E2E production ML pipelines when you're ready.

    Python 359 129 66 45 Updated Apr 24, 2025
  • torchtitan Public

    A PyTorch native library for large-scale model training

    Python 3,626 BSD-3-Clause 343 94 (1 issue needs help) 57 Updated Apr 24, 2025
  • pytorch.github.io Public

    The website for PyTorch

    HTML 252 BSD-3-Clause 300 145 62 Updated Apr 24, 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 91 92 218 (1 issue needs help) 194 Updated Apr 24, 2025
  • benchmark Public

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

    Python 934 BSD-3-Clause 305 78 71 Updated Apr 24, 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 2 2 0 Updated Apr 24, 2025