Skip to content
@pytorch

pytorch

Pinned Loading

  1. pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 90.2k 24.2k

  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.8k 7.1k

  5. torchrec Public

    Pytorch domain library for recommendation systems

    Python 2.2k 516

Repositories

Showing 10 of 82 repositories
  • pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 90,244 24,243 5,000+ (90 issues need help) 1,272 Updated May 24, 2025
  • executorch Public

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

    Python 2,864 551 490 (28 issues need help) 545 Updated May 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 92 97 221 (1 issue needs help) 198 Updated May 24, 2025
  • torchtitan Public

    A PyTorch native platform for training generative AI models

    Python 3,830 BSD-3-Clause 376 100 (1 issue needs help) 62 Updated May 24, 2025
  • TensorRT Public

    PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

    Python 2,758 BSD-3-Clause 365 210 21 Updated May 24, 2025
  • ao Public

    PyTorch native quantization and sparsity for training and inference

    Python 2,057 BSD-3-Clause 267 243 (2 issues need help) 129 Updated May 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 3 2 0 Updated May 24, 2025
  • functorch Public

    functorch is JAX-like composable function transforms for PyTorch.

    Jupyter Notebook 1,424 BSD-3-Clause 104 232 (1 issue needs help) 17 Updated May 24, 2025
  • torchft Public

    PyTorch per step fault tolerance (actively under development)

    Python 301 33 21 (2 issues need help) 15 Updated May 23, 2025
  • xla Public

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

    Python 2,605 527 639 (51 issues need help) 194 Updated May 23, 2025
View all repositories