Skip to content
@pytorch

pytorch

Pinned Loading

  1. pytorch pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 87.2k 23.4k

  2. examples examples Public

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

    Python 22.8k 9.6k

  3. tutorials tutorials Public

    PyTorch tutorials.

    Jupyter Notebook 8.4k 4.1k

  4. vision vision Public

    Datasets, Transforms and Models specific to Computer Vision

    Python 16.6k 7k

  5. torchrec torchrec Public

    Pytorch domain library for recommendation systems

    Python 2k 479

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
  • executorch Public

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

    pytorch/executorch’s past year of commit activity
    C++ 2,532 457 305 (4 issues need help) 574 Updated Feb 25, 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.

    pytorch/test-infra’s past year of commit activity
    TypeScript 88 89 219 (1 issue needs help) 177 Updated Feb 25, 2025
  • TensorRT Public

    PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

    pytorch/TensorRT’s past year of commit activity
    Python 2,683 BSD-3-Clause 360 206 24 Updated Feb 25, 2025
  • ao Public

    PyTorch native quantization and sparsity for training and inference

    pytorch/ao’s past year of commit activity
    Python 1,856 BSD-3-Clause 223 227 (2 issues need help) 111 Updated Feb 25, 2025
  • torchrec Public

    Pytorch domain library for recommendation systems

    pytorch/torchrec’s past year of commit activity
    Python 2,039 BSD-3-Clause 479 63 292 Updated Feb 25, 2025
  • rl Public

    A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

    pytorch/rl’s past year of commit activity
    Python 2,551 MIT 336 173 (4 issues need help) 76 Updated Feb 25, 2025
  • tensordict Public

    TensorDict is a pytorch dedicated tensor container.

    pytorch/tensordict’s past year of commit activity
    Python 886 MIT 81 35 (1 issue needs help) 28 Updated Feb 25, 2025
  • torchtitan Public

    A PyTorch native library for large model training

    pytorch/torchtitan’s past year of commit activity
    Python 3,350 BSD-3-Clause 287 71 (2 issues need help) 50 Updated Feb 25, 2025
  • torchchat Public

    Run PyTorch LLMs locally on servers, desktop and mobile

    pytorch/torchchat’s past year of commit activity
    Python 3,508 BSD-3-Clause 234 79 (6 issues need help) 14 Updated Feb 25, 2025