Skip to content
@pytorch

pytorch

Pinned Loading

  1. pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 89.3k 23.9k

  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
  • 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 23, 2025
  • pytorch Public

    Tensors and Dynamic neural networks in Python with strong GPU acceleration

    Python 89,321 23,949 5,000+ (84 issues need help) 1,312 Updated Apr 23, 2025
  • executorch Public

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

    Python 2,752 528 424 (14 issues need help) 582 Updated Apr 23, 2025
  • xla Public

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

    Python 2,589 506 616 (51 issues need help) 163 Updated Apr 23, 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 46 Updated Apr 23, 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 23, 2025
  • 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 23, 2025
  • pytorch.github.io Public

    The website for PyTorch

    HTML 252 BSD-3-Clause 300 185 62 Updated Apr 23, 2025
  • FBGEMM Public

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

    C++ 1,305 566 41 428 Updated Apr 23, 2025
  • pytorch_sphinx_theme Public

    PyTorch Sphinx Theme

    CSS 99 MIT 139 22 8 Updated Apr 23, 2025
View all repositories