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

    PyTorch native quantization and sparsity for training and inference

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

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

    Python 2,754 528 425 (14 issues need help) 583 Updated Apr 24, 2025
  • pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

  • 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
  • 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,625 BSD-3-Clause 343 94 (1 issue needs help) 57 Updated Apr 24, 2025
  • xla Public

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

    Python 2,589 507 617 (51 issues need help) 164 Updated Apr 24, 2025
  • FBGEMM Public

    FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

    C++ 1,305 566 41 429 Updated Apr 24, 2025
  • pytorch_sphinx_theme Public

    PyTorch Sphinx Theme

    CSS 99 MIT 139 22 8 Updated Apr 24, 2025
  • pytorch.github.io Public

    The website for PyTorch

    HTML 252 BSD-3-Clause 300 145 62 Updated Apr 23, 2025