Skip to content
@pytorch

pytorch

Pinned Loading

  1. pytorch pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 87.8k 23.6k

  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 2.1k 487

Repositories

Showing 10 of 81 repositories
  • xla Public

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

    pytorch/xla’s past year of commit activity
    Python 2,545 505 646 (47 issues need help) 157 Updated Mar 13, 2025
  • 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,586 476 369 (9 issues need help) 581 Updated Mar 13, 2025
  • torchtitan Public

    A PyTorch native library for large model training

    pytorch/torchtitan’s past year of commit activity
    Python 3,442 BSD-3-Clause 309 75 (1 issue needs help) 48 Updated Mar 13, 2025
  • kineto Public

    A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.

    pytorch/kineto’s past year of commit activity
    HTML 778 178 50 9 Updated Mar 13, 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,270 550 38 413 Updated Mar 13, 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 90 89 221 (1 issue needs help) 183 Updated Mar 13, 2025
  • ao Public

    PyTorch native quantization and sparsity for training and inference

    pytorch/ao’s past year of commit activity
    Python 1,898 BSD-3-Clause 228 214 (2 issues need help) 113 Updated Mar 13, 2025
  • tnt Public

    A lightweight library for PyTorch training tools and utilities

    pytorch/tnt’s past year of commit activity
    Python 1,684 283 6 83 Updated Mar 13, 2025
  • torchrec Public

    Pytorch domain library for recommendation systems

    pytorch/torchrec’s past year of commit activity
    Python 2,050 BSD-3-Clause 487 65 304 Updated Mar 13, 2025